implement GetSites()
authorTony Mack <tmack@paris.CS.Princeton.EDU>
Mon, 17 Sep 2012 20:49:10 +0000 (16:49 -0400)
committerTony Mack <tmack@paris.CS.Princeton.EDU>
Mon, 17 Sep 2012 20:49:10 +0000 (16:49 -0400)
PLC/Methods/GetSites.py
PLC/NovaObject.py
PLC/Roles.py
PLC/SiteTags.py
PLC/Sites.py

index d277c28..7095fde 100644 (file)
@@ -13,14 +13,12 @@ class GetSites(Method):
     specified details will be returned.
     """
 
-    roles = ['admin', 'pi', 'user', 'tech', 'node', 'anonymous']
+    roles = ['admin', 'pi', 'user', 'tech', 'anonymous']
 
     accepts = [
         Auth(),
-        Mixed([Mixed(Site.fields['site_id'],
-                     Site.fields['login_base'])],
-              Parameter(str,"login_base"),
-              Parameter(int,"site_id"),
+        Mixed([Site.fields['id']],
+              Parameter(str,"id"),
               Filter(Site.fields)),
         Parameter([str], "List of fields to return", nullok = True)
         ]
@@ -28,4 +26,5 @@ class GetSites(Method):
     returns = [Site.fields]
 
     def call(self, auth, site_filter = None, return_fields = None):
-        return Sites(self.api, site_filter, return_fields)
+        sites = Sites(self.api, site_filter, return_fields)
+        return sites.dicts()
index 382d69f..d4a067d 100644 (file)
@@ -1,3 +1,5 @@
+from PLC.Logger import logger
+
 class NovaObject:
 
     @staticmethod
@@ -15,5 +17,6 @@ class NovaObjects(list):
     def dicts(self):
         result = []
         for obj in self:
+            logger.info(str(obj))
             result.append(NovaObject.object_to_dict(obj, self.fields))
         return result      
index dde6bbf..92cad65 100644 (file)
@@ -29,7 +29,6 @@ class Roles(NovaObjects):
         else:
             raise PLCInvalidArgument, "Wrong role filter %r"%role_filter
 
-        for role in roles:
             #self.append(NovaObject.object_to_dict(role, Role.fields)) 
-            self.append(role
+        self.extend(roles
 
index d04b947..673d739 100644 (file)
@@ -17,14 +17,14 @@ class SiteTag(Row):
     table_name = 'site_tag'
     primary_key = 'site_tag_id'
     fields = {
-        'site_tag_id': Parameter(int, "Site setting identifier"),
-        'site_id': Site.fields['site_id'],
-        'login_base': Site.fields['login_base'],
-        'tag_type_id': TagType.fields['tag_type_id'],
-        'tagname': TagType.fields['tagname'],
-        'description': TagType.fields['description'],
-        'category': TagType.fields['category'],
-        'value': Parameter(str, "Site setting value"),
+        #'site_tag_id': Parameter(int, "Site setting identifier"),
+        #'site_id': Site.fields['site_id'],
+        #'login_base': Site.fields['login_base'],
+        #'tag_type_id': TagType.fields['tag_type_id'],
+        #'tagname': TagType.fields['tagname'],
+        #'description': TagType.fields['description'],
+        #'category': TagType.fields['category'],
+        #'value': Parameter(str, "Site setting value"),
         ### relations
 
         }
index cfb26ad..053c799 100644 (file)
@@ -3,184 +3,29 @@ import string
 
 from PLC.Faults import *
 from PLC.Parameter import Parameter, Mixed
-from PLC.Filter import Filter
-from PLC.Debug import profile
-from PLC.Table import Row, Table
+from PLC.NovaObject import NovaObject, NovaObjects
 from PLC.Slices import Slice, Slices
-from PLC.PCUs import PCU, PCUs
-from PLC.Nodes import Node, Nodes
-from PLC.Addresses import Address, Addresses
-from PLC.Persons import Person, Persons
+#from PLC.Persons import Person, Persons
 
-class Site(Row):
+class Site:
     """
     Representation of a row in the sites table. To use, optionally
     instantiate with a dict of values. Update as you would a
     dict. Commit to the database with sync().
     """
 
-    table_name = 'sites'
-    primary_key = 'site_id'
-    join_tables = ['person_site', 'site_address', 'peer_site']
     fields = {
-        'site_id': Parameter(int, "Site identifier"),
-        'name': Parameter(str, "Full site name", max = 254),
-        'abbreviated_name': Parameter(str, "Abbreviated site name", max = 50),
-        'login_base': Parameter(str, "Site slice prefix", max = 20),
-        'is_public': Parameter(bool, "Publicly viewable site"),
         'enabled': Parameter(bool, "Has been enabled"),
-        'latitude': Parameter(float, "Decimal latitude of the site", min = -90.0, max = 90.0, nullok = True),
-        'longitude': Parameter(float, "Decimal longitude of the site", min = -180.0, max = 180.0, nullok = True),
-        'url': Parameter(str, "URL of a page that describes the site", max = 254, nullok = True),
-        'date_created': Parameter(int, "Date and time when site entry was created, in seconds since UNIX epoch", ro = True),
-        'last_updated': Parameter(int, "Date and time when site entry was last updated, in seconds since UNIX epoch", ro = True),
-        'max_slices': Parameter(int, "Maximum number of slices that the site is able to create"),
-        'max_slivers': Parameter(int, "Maximum number of slivers that the site is able to create"),
+        'id': Parameter(int, "Site identifier"),
+        'name': Parameter(str, "Full site name", max = 254),
+        'description': Parameter(str, "Description", max = 254),
+        #'max_slices': Parameter(int, "Maximum number of slices that the site is able to create"),
+        #'max_slivers': Parameter(int, "Maximum number of slivers that the site is able to create"),
         'person_ids': Parameter([int], "List of account identifiers"),
         'slice_ids': Parameter([int], "List of slice identifiers"),
-        'address_ids': Parameter([int], "List of address identifiers"),
         'pcu_ids': Parameter([int], "List of PCU identifiers"),
         'node_ids': Parameter([int], "List of site node identifiers"),
-        'peer_id': Parameter(int, "Peer to which this site belongs", nullok = True),
-        'peer_site_id': Parameter(int, "Foreign site identifier at peer", nullok = True),
-        'site_tag_ids' : Parameter ([int], "List of tags attached to this site"),
-        'ext_consortium_id': Parameter(int, "external consortium id", nullok = True)
         }
-    related_fields = {
-        'persons': [Mixed(Parameter(int, "Person identifier"),
-                          Parameter(str, "Email address"))],
-        'addresses': [Mixed(Parameter(int, "Address identifer"),
-                            Filter(Address.fields))]
-        }
-    view_tags_name = "view_site_tags"
-    # tags are used by the Add/Get/Update methods to expose tags
-    # this is initialized here and updated by the accessors factory
-    tags = { }
-
-    def validate_name(self, name):
-        if not len(name):
-            raise PLCInvalidArgument, "Name must be specified"
-
-        return name
-
-    validate_abbreviated_name = validate_name
-
-    def validate_login_base(self, login_base):
-        if not len(login_base):
-            raise PLCInvalidArgument, "Login base must be specified"
-
-        if not set(login_base).issubset(string.lowercase + string.digits):
-            raise PLCInvalidArgument, "Login base must consist only of lowercase ASCII letters or numbers"
-
-        conflicts = Sites(self.api, [login_base])
-        for site in conflicts:
-            if 'site_id' not in self or self['site_id'] != site['site_id']:
-                raise PLCInvalidArgument, "login_base already in use"
-
-        return login_base
-
-    def validate_latitude(self, latitude):
-        if not self.has_key('longitude') or \
-           self['longitude'] is None:
-            raise PLCInvalidArgument, "Longitude must also be specified"
-
-        return latitude
-
-    def validate_longitude(self, longitude):
-        if not self.has_key('latitude') or \
-           self['latitude'] is None:
-            raise PLCInvalidArgument, "Latitude must also be specified"
-
-        return longitude
-
-    validate_date_created = Row.validate_timestamp
-    validate_last_updated = Row.validate_timestamp
-
-    add_person = Row.add_object(Person, 'person_site')
-    remove_person = Row.remove_object(Person, 'person_site')
-
-    add_address = Row.add_object(Address, 'site_address')
-    remove_address = Row.remove_object(Address, 'site_address')
-
-    def update_last_updated(self, commit = True):
-        """
-        Update last_updated field with current time
-        """
-
-        assert 'site_id' in self
-        assert self.table_name
-
-        self.api.db.do("UPDATE %s SET last_updated = CURRENT_TIMESTAMP " % (self.table_name) + \
-                       " where site_id = %d" % (self['site_id']) )
-        self.sync(commit)
-
-
-    def associate_persons(self, auth, field, value):
-        """
-        Adds persons found in value list to this site (using AddPersonToSite).
-        Deletes persons not found in value list from this site (using DeletePersonFromSite).
-        """
-
-        assert 'person_ids' in self
-        assert 'site_id' in self
-        assert isinstance(value, list)
-
-        (person_ids, emails) = self.separate_types(value)[0:2]
-
-        # Translate emails into person_ids
-        if emails:
-            persons = Persons(self.api, emails, ['person_id']).dict('person_id')
-            person_ids += persons.keys()
-
-        # Add new ids, remove stale ids
-        if self['person_ids'] != person_ids:
-            from PLC.Methods.AddPersonToSite import AddPersonToSite
-            from PLC.Methods.DeletePersonFromSite import DeletePersonFromSite
-            new_persons = set(person_ids).difference(self['person_ids'])
-            stale_persons = set(self['person_ids']).difference(person_ids)
-
-            for new_person in new_persons:
-                AddPersonToSite.__call__(AddPersonToSite(self.api), auth, new_person, self['site_id'])
-            for stale_person in stale_persons:
-                DeletePersonFromSite.__call__(DeletePersonFromSite(self.api), auth, stale_person, self['site_id'])
-
-    def associate_addresses(self, auth, field, value):
-        """
-        Deletes addresses_ids not found in value list (using DeleteAddress).
-        Adds address if slice_fields w/o address_id found in value list (using AddSiteAddress).
-        Update address if slice_fields w/ address_id found in value list (using UpdateAddress).
-        """
-
-        assert 'address_ids' in self
-        assert 'site_id' in self
-        assert isinstance(value, list)
-
-        (address_ids, blank, addresses) = self.separate_types(value)
-
-        for address in addresses:
-            if 'address_id' in address:
-                address_ids.append(address['address_id'])
-
-        # Add new ids, remove stale ids
-        if self['address_ids'] != address_ids:
-            from PLC.Methods.DeleteAddress import DeleteAddress
-            stale_addresses = set(self['address_ids']).difference(address_ids)
-
-            for stale_address in stale_addresses:
-                DeleteAddress.__call__(DeleteAddress(self.api), auth, stale_address)
-
-        if addresses:
-            from PLC.Methods.AddSiteAddress import AddSiteAddress
-            from PLC.Methods.UpdateAddress import UpdateAddress
-
-            updated_addresses = filter(lambda address: 'address_id' in address, addresses)
-            added_addresses = filter(lambda address: 'address_id' not in address, addresses)
-
-            for address in added_addresses:
-                AddSiteAddress.__call__(AddSiteAddress(self.api), auth, self['site_id'], address)
-            for address in updated_addresses:
-                address_id = address.pop('address_id')
-                UpdateAddress.__call__(UpdateAddress(self.api), auth, address_id, address)
 
     def delete(self, commit = True):
         """
@@ -233,40 +78,24 @@ class Site(Row):
         self['deleted'] = True
         self.sync(commit)
 
-class Sites(Table):
+class Sites(NovaObjects):
     """
     Representation of row(s) from the sites table in the
     database.
     """
 
     def __init__(self, api, site_filter = None, columns = None):
-        Table.__init__(self, api, Site, columns)
-
-        view = "view_sites"
-        for tagname in self.tag_columns:
-            view= "%s left join %s using (%s)"%(view,Site.tagvalue_view_name(tagname),
-                                                Site.primary_key)
-
-        sql = "SELECT %s FROM %s WHERE deleted IS False" % \
-            (", ".join(self.columns.keys()+self.tag_columns.keys()),view)
-
-        if site_filter is not None:
-            if isinstance(site_filter, (list, tuple, set)):
-                # Separate the list into integers and strings
-                ints = filter(lambda x: isinstance(x, (int, long)), site_filter)
-                strs = filter(lambda x: isinstance(x, StringTypes), site_filter)
-                site_filter = Filter(Site.fields, {'site_id': ints, 'login_base': strs})
-                sql += " AND (%s) %s" % site_filter.sql(api, "OR")
-            elif isinstance(site_filter, dict):
-                site_filter = Filter(Site.fields, site_filter)
-                sql += " AND (%s) %s" % site_filter.sql(api, "AND")
-            elif isinstance (site_filter, StringTypes):
-                site_filter = Filter(Site.fields, {'login_base':site_filter})
-                sql += " AND (%s) %s" % site_filter.sql(api, "AND")
-            elif isinstance (site_filter, (int, long)):
-                site_filter = Filter(Site.fields, {'site_id':site_filter})
-                sql += " AND (%s) %s" % site_filter.sql(api, "AND")
-            else:
-                raise PLCInvalidArgument, "Wrong site filter %r"%site_filter
-
-        self.selectall(sql)
+        self.api = api 
+        self.fields = Site.fields
+        if not site_filter:
+            sites = self.api.client_shell.keystone.tenants.findall()
+        elif isintance(site_filter, StringTypes):
+            sites = [self.api.client_shell.keystone.tenants.find(id=site_filter)]
+        elif isintance(site_filter, StringTypes):
+            sites = [self.api.client_shell.keystone.tenants.find(**site_filter)]
+        elif isinstance(site_filter, (list, tuple, set)):
+            sites = self.api.client_shell.keystone.tenants.findall()
+            sites = [site for site in sites if site.id in site_filter]
+        else:
+            raise PLCInvalidArgument, "Wrong site filter %s" % site_filter         
+        self.extend(sites)