-from types import StringTypes
import string
from PLC.Faults import *
-from PLC.Parameter import Parameter
+from PLC.Parameter import Parameter, Mixed
+from PLC.Filter import Filter
from PLC.Debug import profile
from PLC.Table import Row, Table
from PLC.Slices import Slice, Slices
from PLC.PCUs import PCU, PCUs
from PLC.Nodes import Node, Nodes
-from PLC.NodeGroups import NodeGroup, NodeGroups
from PLC.Addresses import Address, Addresses
-import PLC.Persons
+from PLC.Persons import Person, Persons
class Site(Row):
"""
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),
+ 'login_base': Parameter(str, "Site slice prefix", max = 32),
'is_public': Parameter(bool, "Publicly viewable site"),
- 'latitude': Parameter(float, "Decimal latitude of the site", min = -90.0, max = 90.0),
- 'longitude': Parameter(float, "Decimal longitude of the site", min = -180.0, max = 180.0),
- 'url': Parameter(str, "URL of a page that describes the site", max = 254),
+ '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"),
- 'person_ids': Parameter([int], "List of account identifiers", ro = True),
- 'slice_ids': Parameter([int], "List of slice identifiers", ro = True),
- 'address_ids': Parameter([int], "List of address identifiers", ro = True),
- 'pcu_ids': Parameter([int], "List of PCU identifiers", ro = True),
- 'node_ids': Parameter([int], "List of site node identifiers", ro = True),
+ '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"
+ raise PLCInvalidArgument("Name must be specified")
return name
def validate_login_base(self, login_base):
if not len(login_base):
- raise PLCInvalidArgument, "Login base must be specified"
+ raise PLCInvalidArgument("Login base must be specified")
- if not set(login_base).issubset(string.ascii_letters.lower()):
- raise PLCInvalidArgument, "Login base must consist only of lowercase ASCII letters"
+ if not set(login_base).issubset(string.ascii_lowercase + string.digits + '.'):
+ raise PLCInvalidArgument("Login base must consist only of lowercase ASCII letters or numbers or dots")
conflicts = Sites(self.api, [login_base])
- for site_id, site in conflicts.iteritems():
- if 'site_id' not in self or self['site_id'] != site_id:
- raise PLCInvalidArgument, "login_base already in use"
+ 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 \
+ if 'longitude' not in self or \
self['longitude'] is None:
- raise PLCInvalidArgument, "Longitude must also be specified"
+ raise PLCInvalidArgument("Longitude must also be specified")
return latitude
def validate_longitude(self, longitude):
- if not self.has_key('latitude') or \
+ if 'latitude' not in self or \
self['latitude'] is None:
- raise PLCInvalidArgument, "Latitude must also be specified"
+ raise PLCInvalidArgument("Latitude must also be specified")
return longitude
- def add_person(self, person, commit = True):
- """
- Add person to existing site.
- """
-
- assert 'site_id' in self
- assert isinstance(person, PLC.Persons.Person)
- assert 'person_id' in person
-
- site_id = self['site_id']
- person_id = person['person_id']
-
- if person_id not in self['person_ids']:
- assert site_id not in person['site_ids']
-
- self.api.db.do("INSERT INTO person_site (person_id, site_id)" \
- " VALUES(%(person_id)d, %(site_id)d)",
- locals())
+ validate_date_created = Row.validate_timestamp
+ validate_last_updated = Row.validate_timestamp
- if commit:
- self.api.db.commit()
+ add_person = Row.add_object(Person, 'person_site')
+ remove_person = Row.remove_object(Person, 'person_site')
- self['person_ids'].append(person_id)
- person['site_ids'].append(site_id)
+ add_address = Row.add_object(Address, 'site_address')
+ remove_address = Row.remove_object(Address, 'site_address')
- def remove_person(self, person, commit = True):
+ def update_last_updated(self, commit = True):
"""
- Remove person from existing site.
+ Update last_updated field with current time
"""
assert 'site_id' in self
- assert isinstance(person, PLC.Persons.Person)
- assert 'person_id' in person
+ assert self.table_name
- site_id = self['site_id']
- person_id = person['person_id']
-
- if person_id in self['person_ids']:
- assert site_id in person['site_ids']
-
- self.api.db.do("DELETE FROM person_site" \
- " WHERE person_id = %(person_id)d" \
- " AND site_id = %(site_id)d",
- locals())
-
- if commit:
- self.api.db.commit()
+ self.api.db.do("UPDATE %s SET last_updated = CURRENT_TIMESTAMP " % (self.table_name) + \
+ " where site_id = %d" % (self['site_id']) )
+ self.sync(commit)
- self['person_ids'].remove(person_id)
- person['site_ids'].remove(site_id)
- def add_address(self, address, commit = True):
+ def associate_persons(self, auth, field, value):
"""
- Add address to existing site.
+ 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(address, Address)
- assert 'address_id' in address
+ assert isinstance(value, list)
- site_id = self['site_id']
- address_id = address['address_id']
+ (person_ids, emails) = self.separate_types(value)[0:2]
- if address_id not in self['address_ids']:
- self.api.db.do("INSERT INTO site_address (address_id, site_id)" \
- " VALUES(%(address_id)d, %(site_id)d)",
- locals())
+ # Translate emails into person_ids
+ if emails:
+ persons = Persons(self.api, emails, ['person_id']).dict('person_id')
+ person_ids += list(persons.keys())
- if commit:
- self.api.db.commit()
+ # 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)
- self['address_ids'].append(address_id)
+ 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 remove_address(self, address, commit = True):
+ def associate_addresses(self, auth, field, value):
"""
- Remove address from existing site.
+ 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(address, Address)
- assert 'address_id' in address
+ 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)
- site_id = self['site_id']
- address_id = address['address_id']
+ for stale_address in stale_addresses:
+ DeleteAddress.__call__(DeleteAddress(self.api), auth, stale_address)
- if address_id in self['address_ids']:
- self.api.db.do("DELETE FROM site_address" \
- " WHERE address_id = %(address_id)d" \
- " AND site_id = %(site_id)d",
- locals())
+ if addresses:
+ from PLC.Methods.AddSiteAddress import AddSiteAddress
+ from PLC.Methods.UpdateAddress import UpdateAddress
- if commit:
- self.api.db.commit()
+ updated_addresses = [address for address in addresses if 'address_id' in address]
+ added_addresses = [address for address in addresses if 'address_id' not in address]
- self['address_ids'].remove(address_id)
+ 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):
"""
# Delete accounts of all people at the site who are not
# members of at least one other non-deleted site.
- persons = PLC.Persons.Persons(self.api, self['person_ids'])
- for person_id, person in persons.iteritems():
+ persons = Persons(self.api, self['person_ids'])
+ for person in persons:
delete = True
person_sites = Sites(self.api, person['site_ids'])
- for person_site_id, person_site in person_sites.iteritems():
- if person_site_id != self['site_id']:
+ for person_site in person_sites:
+ if person_site['site_id'] != self['site_id']:
delete = False
break
# Delete all site addresses
addresses = Addresses(self.api, self['address_ids'])
- for address in addresses.values():
- address.delete(commit = False)
+ for address in addresses:
+ address.delete(commit = False)
# Delete all site slices
slices = Slices(self.api, self['slice_ids'])
- for slice in slices.values():
- slice.delete(commit = False)
+ for slice in slices:
+ slice.delete(commit = False)
# Delete all site PCUs
pcus = PCUs(self.api, self['pcu_ids'])
- for pcu in pcus.values():
- pcu.delete(commit = False)
+ for pcu in pcus:
+ pcu.delete(commit = False)
# Delete all site nodes
nodes = Nodes(self.api, self['node_ids'])
- for node in nodes.values():
+ for node in nodes:
node.delete(commit = False)
# Clean up miscellaneous join tables
- for table in ['person_site']:
- self.api.db.do("DELETE FROM %s" \
- " WHERE site_id = %d" % \
- (table, self['site_id']), self)
+ for table in self.join_tables:
+ self.api.db.do("DELETE FROM %s WHERE site_id = %d" % \
+ (table, self['site_id']))
# Mark as deleted
self['deleted'] = True
class Sites(Table):
"""
Representation of row(s) from the sites table in the
- database. Specify fields to limit columns to just the specified
- fields.
+ database.
"""
- def __init__(self, api, site_id_or_login_base_list = None):
- self.api = api
-
- sql = "SELECT %s FROM view_sites WHERE deleted IS False" % \
- ", ".join(Site.fields)
-
- if site_id_or_login_base_list:
- # Separate the list into integers and strings
- site_ids = filter(lambda site_id: isinstance(site_id, (int, long)),
- site_id_or_login_base_list)
- login_bases = filter(lambda login_base: isinstance(login_base, StringTypes),
- site_id_or_login_base_list)
- sql += " AND (False"
- if site_ids:
- sql += " OR site_id IN (%s)" % ", ".join(map(str, site_ids))
- if login_bases:
- sql += " OR login_base IN (%s)" % ", ".join(api.db.quote(login_bases))
- sql += ")"
-
- rows = self.api.db.selectall(sql)
-
- for row in rows:
- self[row['site_id']] = site = Site(api, row)
- for aggregate in ['person_ids', 'slice_ids', 'address_ids',
- 'pcu_ids', 'node_ids']:
- if not site.has_key(aggregate) or site[aggregate] is None:
- site[aggregate] = []
- else:
- site[aggregate] = map(int, site[aggregate].split(','))
+ 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(list(self.columns.keys())+list(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 = [x for x in site_filter if isinstance(x, int)]
+ strs = [x for x in site_filter if isinstance(x, str)]
+ 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):
+ allowed_fields=dict(list(Site.fields.items())+list(Site.tags.items()))
+ site_filter = Filter(allowed_fields, site_filter)
+ sql += " AND (%s) %s" % site_filter.sql(api, "AND")
+ elif isinstance (site_filter, str):
+ site_filter = Filter(Site.fields, {'login_base':site_filter})
+ sql += " AND (%s) %s" % site_filter.sql(api, "AND")
+ elif isinstance (site_filter, int):
+ 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)