# # Thierry Parmentelat - INRIA # import re from types import StringTypes from PLC.Faults import * from PLC.Parameter import Parameter from PLC.Filter import Filter from PLC.Table import Row, Table class Peer (Row): """ Stores the list of peering PLCs in the peers table. See the Row class for more details """ table_name = 'peers' primary_key = 'peer_id' fields = { 'peer_id' : Parameter (int, "Peer identifier"), 'peername' : Parameter (str, "Peer name"), 'peer_url' : Parameter (str, "Peer API url"), 'person_id' : Parameter (int, "Person_id of the account storing credentials - temporary"), 'node_ids' : Parameter ([int], "This peer's nodes ids") } def validate_peer_url (self, url): """ Validate URL, checks it looks like https """ invalid_url = PLCInvalidArgument("Invalid URL") if not re.compile ("^https://.*$").match(url) : raise invalid_url return url def delete (self, commit=True): """ Delete peer """ assert 'peer_id' in self self['deleted'] = True self.sync(commit) class Peers (Table): """ Maps to the peers table in the database """ def __init__ (self, api, peer_filter = None, columns = None): Table.__init__(self, api, Peer, columns) sql = "SELECT %s FROM view_peers WHERE deleted IS False" % \ ", ".join(self.columns) if peer_filter is not None: if isinstance(peer_filter, (list, tuple, set)): # Separate the list into integers and strings ints = filter(lambda x: isinstance(x, (int, long)), peer_filter) strs = filter(lambda x: isinstance(x, StringTypes), peer_filter) peer_filter = Filter(Peer.fields, {'peer_id': ints, 'peername': strs}) sql += " AND (%s)" % peer_filter.sql(api, "OR") elif isinstance(peer_filter, dict): peer_filter = Filter(Peer.fields, peer_filter) sql += " AND (%s)" % peer_filter.sql(api, "AND") self.selectall(sql)