def intersect(self, other):
"""See RFC 3264."""
map1 = self.rtpmap
d1 = {}
for code,(e,fmt) in map1.items():
d1[rtpmap2canonical(code,e)] = e
map2 = other.rtpmap
outmap = OrderedDict()
# XXX quadratic - make rtpmap an ordereddict
for code, (e, fmt) in map2.items():
canon = rtpmap2canonical(code,e)
if d1.has_key(canon):
outmap[code] = (e, fmt)
self.rtpmap = outmap
self.formats = [ str(x) for x in self.rtpmap.keys() ]
self._a['rtpmap'] = OrderedDict([ (code,e) for
(code, (e, fmt)) in outmap.items() ])
def load_from_persist(self, restart = False):
self.clear_sa()
self.tables = OrderedDict()
try:
self.code_repr_load()
except ImportError:
return
self.add_relations()
self.update_sa()
self.validate_database()
def dump_supported_features():
import drivers
features = OrderedDict()
for nbr, driver_features in enumerate([drivers.DriverBase, drivers.pysqlite2,
drivers.MySQLdb, drivers.psycopg2, drivers.ibm_db,
drivers.cx_Oracle, drivers.ceODBC]):
for feature in dir(driver_features):
if (feature[:1] == '_' or
feature == 'typemap' or
hasattr(getattr(driver_features, feature), '__call__')):
continue # Skip private attributes & methods
value = getattr(driver_features, feature)
(doc, supports) = features.get(feature, ['', []])
if not doc:
doc = getattr(value, '__doc__')
supports.append(value)
features[feature] = (doc, supports)
#Output the summary & introduction section
print """#summary Comparison of core features provided by several DB-API implementations. \
Covers features that are universal to all DB-API implementations; features that are specific \
to a particular driver are not covered. \
= Introduction =
Comparison of core features provided by several DB-API implementations. \
Covers features that are universal to all DB-API implementations ; features that are specific \
to a particular driver are not covered.
"""
#Output the table headers
print ("|| Feature || Feature Description || Default Value || pysqlite2 || MySQLdb "
"|| psycopg2 || ibmdb || ceODBC || cx_Oracle")
#Output the table details
for (feature, (doc, supports)) in features.items():
print '||', feature, ' || ', doc, ' || ',
for support in supports:
print support, ' ||',
print
def dump_testcase_reports():
qualfiles = glob.glob("reports/*")
files = [qualfile[qualfile.find('/')+1:] for qualfile in qualfiles]
print '|| Test Name ||','||'.join(files),'||'
tests = OrderedDict()
for qualfile in qualfiles:
# Add each test to "tests" dict
for test_result in open(qualfile, 'r').readlines():
break_loc = test_result.find('...')
test, result = None, None
test = test_result[:break_loc - 1]
result = test_result[break_loc + 4:-1]
if (not break_loc) or test.startswith('----') or test.startswith('Ran '):
continue
if result == "ok":
result = "Pass"
if result == "(Unsupported) ok":
result = "Fail"
if result == "(Skipped) ok":
result = "N/A"
# Append this test result to list of all results for this test
results = tests.get(test, [])
results.append(result)
tests[test] = results
##print '||', test, '||', result
for (key, value) in tests.items():
if key.strip()=='O' or value == ['N/A', 'N/A', 'N/A']:
continue
print '||', key, '||', ' || '.join(value), '||'
def __init__(self, text=None):
self.nettype = 'IN'
self.addrfamily = 'IP4'
self.ipaddr = None
self.port = None
self.transport = None
self.formats = []
self._d = {}
self._a = {}
self.rtpmap = OrderedDict()
self.media = 'audio'
self.transport = 'RTP/AVP'
self.keyManagement = None
if text:
parse_m(self, 'm', text)
def __init__(self, *tables, **kw):
log.info("initialising database")
self.status = "updating"
self.kw = kw
self.metadata = None
self.connection_string = kw.get("connection_string", None)
if self.connection_string:
self.engine = create_engine(self.connection_string)
self.metadata = MetaData()
self.metadata.bind = self.engine
self._Session = sessionmaker(bind=self.engine, autoflush = False)
self.Session = sessionwrapper.SessionClass(self._Session, self)
self.logging_tables = kw.get("logging_tables", None)
self.quiet = kw.get("quiet", None)
self.application = kw.get("application", None)
if self.application:
self.set_application(self.application)
self.max_table_id = 0
self.max_event_id = 0
self.persisted = False
self.graph = None
self.relations = []
self.tables = OrderedDict()
self.search_actions = {}
self.search_names = {}
self.search_ids = {}
for table in tables:
self.add_table(table)
title = title.decode("utf-8")
dyna = dyna.decode("utf-8")
auth = auth.decode("utf-8")
o = by_title.get(title, None)
if o is None:
print title
continue
links = o[u"資料"]
if any((link[u"標題"] == NAME) for link in links):
continue
d = OrderedDict()
d[u"標題"] = NAME
d[u"網址"] = u"https://www.kanripo.org" + url
d[u"限制"] = u"无"
d[u"文字"] = True
d[u"影印"] = False
d[u"简体"] = False
links.append(d)
if o[u"朝代"] is None:
if dyna:
o[u"朝代"] = dyna
if o[u"作者"] is None:
if auth:
o[u"作者"] = auth
def reload_rules(self):
self.rules = OrderedDict()
self.load_rules()
def clearRtpMap(self):
self.rtpmap = OrderedDict()
def __reginit__(self):
self.rules = OrderedDict()
self.load_rules()
class MediaDescription:
"The MediaDescription encapsulates all of the SDP media descriptions"
def __init__(self, text=None):
self.nettype = 'IN'
self.addrfamily = 'IP4'
self.ipaddr = None
self.port = None
self.transport = None
self.formats = []
self._d = {}
self._a = {}
self.rtpmap = OrderedDict()
self.media = 'audio'
self.transport = 'RTP/AVP'
self.keyManagement = None
if text:
parse_m(self, 'm', text)
def setFormats(self, formats):
if self.media in ( 'audio', 'video'):
for pt in formats:
pt = int(pt)
if pt < 97:
try:
PT = RTPDict[pt]
except KeyError:
# We don't know this one - hopefully there's an
# a=rtpmap entry for it.
continue
self.addRtpMap(PT)
# XXX the above line is unbound local variable error if
# not RTPDict.has_key(pt) --Zooko 2004-09-29
self.formats = formats
def setMedia(self, media):
"""Set the media type.
@param media: must be 'audio' or 'video'
"""
self.media = media
def setTransport(self, transport):
self.transport = transport
def setServerIP(self, l):
self.ipaddr = l
def setLocalPort(self, l):
self.port = l
def setKeyManagement(self, km):
parse_a(self, 'keymgmt', km)
def clearRtpMap(self):
self.rtpmap = OrderedDict()
def addRtpMap(self, fmt):
if fmt.pt is None:
pts = self.rtpmap.keys()
pts.sort()
if pts and pts[-1] > 100:
payload = pts[-1] + 1
else:
payload = 101
else:
payload = fmt.pt
rtpmap = "%d %s/%d%s%s"%(payload, fmt.name, fmt.clock,
((fmt.params and '/') or ""),
fmt.params or "")
self.rtpmap[int(payload)] = (rtpmap, fmt)
self._a.setdefault('rtpmap', OrderedDict())[payload] = rtpmap
self.formats.append(str(payload))
def intersect(self, other):
"""See RFC 3264."""
map1 = self.rtpmap
d1 = {}
for code,(e,fmt) in map1.items():
d1[rtpmap2canonical(code,e)] = e
map2 = other.rtpmap
outmap = OrderedDict()
# XXX quadratic - make rtpmap an ordereddict
for code, (e, fmt) in map2.items():
canon = rtpmap2canonical(code,e)
if d1.has_key(canon):
outmap[code] = (e, fmt)
self.rtpmap = outmap
self.formats = [ str(x) for x in self.rtpmap.keys() ]
self._a['rtpmap'] = OrderedDict([ (code,e) for
(code, (e, fmt)) in outmap.items() ])
class Table(object):
""" this holds metadata relating to a database table. It also
contains methods to create a sqlalchemy Table,Mapper and Class."""
modifiable_kw = ["quiet", "lookup", "table_class", "primary_entities",
"secondary_entities", "logged", "validated",
"table_type", "title_field", "description_field",
"summary_fields", "summary", "default_node",
"valid_info_tables"]
restricted_kw = ["table_id", "field_order", "persisted",
"primary_key", "entity", "relation", "info_table",
"modified_by", "modified_date", "version", "entity_relationship"]
def __init__(self, name, *args , **kw):
"""
name: name of the table
primary_key: a comma delimited string stating what field
should act as a primary key.
logged: Boolean stating if the table should be logged
modified_by: Boolean stating if the table should have a last
modified by
*args : All the Field objects this table has.
"""
self.all_kw = set(self.modifiable_kw + self.restricted_kw)
self.all_updatable_kw = set(self.modifiable_kw)
assert set(kw.keys()) - self.all_kw == set()
self.name = name
self.kw = kw
self.args = args
self.table_id = kw.get("table_id", None)
self.field_list = []
self.fields = OrderedDict()
self.field_order = kw.get("field_order", [])
self.current_order = 0
self.primary_key = kw.get("primary_key", None)
#persisted should be private
self.persisted = kw.get("persisted", False)
self.quiet = kw.get("quiet", False)
# table types
self.entity = kw.get("entity", False)
self.relation = kw.get("relation", False)
self.lookup = kw.get("lookup", False)
self.info_table = kw.get("info_table", False)
self.table_class = kw.get("table_class")
## for relaion tables
primary_entities = kw.get("primary_entities")
if primary_entities :
self.primary_entities = primary_entities.strip().split(",")
secondary_entities = kw.get("secondary_entities")
if secondary_entities:
self.secondary_entities = secondary_entities.strip().split(",")
## for entity or relation tables
valid_info_tables = kw.get("valid_info_tables")
if valid_info_tables:
self.valid_info_tables = valid_info_tables.split()
else:
self.valid_info_tables = []
## for info tables to be populated
self.valid_core_types = []
self.max_field_id = 0
self.logged = kw.get("logged", True)
self.validated = kw.get("validated", True)
self.modified_date = kw.get("modified_date", True)
self.modified_by = kw.get("modified_by", True)
self.version = kw.get("version", True)
self.table_type = kw.get("table_type", "user")
self.title_field = kw.get("title_field", None)
self.description_field = kw.get("description_field", None)
self.summary_fields = kw.get("summary_fields", None)
self.summary = kw.get("summary", None)
self.default_node = kw.get("default_node", None)
self.primary_key_list = []
if self.primary_key:
self.primary_key_list = self.primary_key.split(",")
self.events = dict(new = [],
delete = [],
change = [])
for field in args:
field._set_parent(self)
if "_version" not in self.fields and self.version:
self.add_field(Integer("_version"))
if "_modified_date" not in self.fields and self.modified_date:
self.add_field(Modified("_modified_date"))
if "_modified_by" not in self.fields and self.modified_by:
self.add_field(ModifiedBySession("_modified_by"))
self.foriegn_key_columns_current = None
#sqlalchemy objects
self.sa_table = None
self.sa_class = None
self.mapper = None
self.paths = None
self.local_tables = None
self.one_to_many_tables = None
self.table_path_list = None
self.table_path = None
self.schema_dict = None
self.schema = None
self.columns_cache = None
def code_repr(self):
header = "Table('%s',\n " % self.name
footer = "\n)"
field_list = [field.code_repr() for field in
sorted(self.field_list, key = lambda x:x.field_id)]
fields_repr = ",\n ".join(field_list)
kw_repr = ""
kw_list = ["%s = %s" % (i[0], repr(i[1])) for i in self.kw.items()]
if self.kw:
kw_repr = ",\n " + ",\n ".join(sorted(kw_list))
return header + fields_repr + kw_repr + footer
def __repr__(self):
return "%s - %s" % (self.name, self.columns.keys())
def set_kw(self, key, value):
if key not in self.all_updatable_kw:
raise ValueError("%s not allowed to be added or modified" % key)
with SchemaLock(self.database) as file_lock:
setattr(self, key, value)
self.kw[key] = value
file_lock.export()
def add_foreign_key_columns(self):
for column in self.foriegn_key_columns.values():
original_col = column.original_column
name = column.name
if original_col == "id" and name not in self.defined_columns:
relation = column.defined_relation
field = relation.parent
new_field = Integer(name, mandatory = relation.many_side_not_null,
default = relation.many_side_default,
onupdate = relation.many_side_onupdate)
self._add_field_no_persist(new_field)
field.kw["foreign_key_name"] = name
field.foreign_key_name = name
def add_relation(self, field):
#TODO make into more general add relation, currently
#only works if both tables are already persisted
if not self.persisted:
self._add_field_no_persist(field)
return
with SchemaLock(self.database) as file_lock:
self._add_field_no_persist(field)
self.database.add_relations()
name, relation = field.relations.copy().popitem()
fk_table = self.database[relation.foreign_key_table]
pk_table = self.database[relation.primary_key_table]
for name, column in fk_table.foriegn_key_columns.iteritems():
if column.defined_relation.parent == field:
sa_options = column.sa_options
## sqlite rubbish
if self.database.engine.name == 'sqlite':
sa_options["server_default"] = "null"
col = sa.Column(name, column.type, **sa_options)
fk_table.add_foreign_key_columns()
file_lock.export(uuid = True)
print "hererer"
col.create(fk_table.sa_table)
for name, con in fk_table.foreign_key_constraints.iteritems():
if self.database.engine.name == 'sqlite':
break
fk_const = migrate.changeset.constraint.ForeignKeyConstraint(con[0],
con[1], name = name, table = fk_table.sa_table)
if name == relation.foreign_key_constraint_name:
fk_const.create()
file_lock.export()
self.database.load_from_persist(True)
def delete_relation(self, field):
if isinstance(field, basestring):
field = self.fields[field]
name, relation = field.relations.copy().popitem()
with SchemaLock(self.database) as file_lock:
mandatory = True if relation.many_side_not_null else False
fk_table = self.database[relation.foreign_key_table]
pk_table = self.database[relation.primary_key_table]
field.table.fields.pop(field.name)
file_lock.export(uuid = True)
for name, con in fk_table.foreign_key_constraints.iteritems():
fk_const = migrate.changeset.constraint.ForeignKeyConstraint(con[0],
con[1], name = name, table = fk_table.sa_table)
if name == relation.foreign_key_constraint_name:
fk_const.drop()
file_lock.export()
self.database.load_from_persist(True)
def add_index(self, field, defer_update_sa = False):
with SchemaLock(self.database) as file_lock:
self._add_field_no_persist(field)
file_lock.export(uuid = True)
name, index = field.indexes.popitem()
ind = [self.sa_table.columns[col.strip()] for col in index.fields.split(",")]
if index.type == "unique":
sa.Index(index.name, *ind, unique = True).create()
else:
sa.Index(index.name, *ind).create()
file_lock.export()
self.database.load_from_persist(True)
def delete_index(self, field):
if isinstance(field, basestring):
field = self.fields[field]
with SchemaLock(self.database) as file_lock:
field.table.fields.pop(field.name)
file_lock.export(uuid = True)
for sa_index in self.sa_table.indexes:
if sa_index.name == field.name:
sa_index.drop()
file_lock.export()
self.database.load_from_persist(True)
def add_field(self, field, defer_update_sa = False):
"""add a Field object to this Table"""
if self.persisted == True:
with SchemaLock(self.database) as file_lock:
self._add_field_no_persist(field)
file_lock.export(uuid = True)
self._add_field_by_alter_table(field)
file_lock.export()
self.database.load_from_persist(True)
else:
self._add_field_no_persist(field)
def add_event(self, event):
self.add_field(event)
def rename_field(self, field, new_name):
if self.database.engine.name == "sqlite":
##FIXME make better exception
raise Exception("sqlite cannot alter fields")
if isinstance(field, basestring):
field = self.fields[field]
with SchemaLock(self.database) as file_lock:
field.name = new_name
column = field.columns[field.column_order[0]] ##TODO make sure only 1 column in field
file_lock.export(uuid = True)
self.sa_table.c[column.name].alter(name = new_name)
file_lock.export()
self.database.load_from_persist(True)
def drop_field(self, field):
if self.database.engine.name == "sqlite":
##FIXME make better exception
raise Exception("sqlite cannot alter fields")
if isinstance(field, basestring):
field = self.fields[field]
with SchemaLock(self.database) as file_lock:
self.fields.pop(field.name)
file_lock.export(uuid = True)
for column in field.columns.values():
self.sa_table.c[column.name].drop()
file_lock.export()
self.database.load_from_persist(True)
def alter_field(self, field, **kw):
if isinstance(field, basestring):
field = self.fields[field]
if field.category <> "field":
raise Exception(("only fields representing database"
"fields can be altered"))
connection = self.database.db.open()
with SchemaLock(self.database) as file_lock:
field_type = kw.pop("type", None)
field.kw.update(kw)
if field_type:
if isinstance(field_type, basestring):
field_type = getattr(fields, field_type)
new_field = field_type(field.name, **field.kw)
else:
new_field = field.__class__(field.name, **field.kw)
_, column = new_field.columns.copy().popitem()
self.fields[field.name] = new_field
file_lock.export(uuid = True)
sa_options = column.sa_options
col = self.sa_table.c[column.name]
col.alter(sa.Column(column.name, column.type, **sa_options))
file_lock.export()
self.database.load_from_persist(True)
def _add_field_no_persist(self, field):
"""add a Field object to this Table"""
field._set_parent(self)
def _add_field_by_alter_table(self, field):
for name, column in field.columns.iteritems():
col = sa.Column(name, column.type, **column.sa_options)
col.create(self.sa_table)
@property
def ordered_fields(self):
fields = []
for field in self.field_order:
rfield = self.fields[field]
if rfield.category in ("internal", "multi_field", "field"):
fields.append(rfield)
return fields
@property
def ordered_user_fields(self):
return filter(lambda f: f.category.endswith("field"),
self.ordered_fields)
@property
def items(self):
"""gathers all columns and relations defined in this table"""
items = {}
for fields in self.fields.itervalues():
for name, item in fields.items.iteritems():
items[name] = item
return items
@property
def defined_columns(self):
"""gathers all columns defined in this table"""
columns = {}
for field in self.fields.itervalues():
for name, column in field.columns.iteritems():
columns[name] = column
return columns
@property
def indexes(self):
"""gathers all columns defined in this table"""
columns = {}
for field in self.fields.itervalues():
for name, column in field.indexes.iteritems():
columns[name] = column
return columns
@property
def constraints(self):
"""gathers all columns defined in this table"""
columns = {}
for field in self.fields.itervalues():
for name, column in field.constraints.iteritems():
columns[name] = column
return columns
@property
def defined_columns_order(self):
column_order = []
for field in self.field_order:
for column in self.fields[field].column_order:
column_order.append(column)
return column_order
@property
def columns(self):
"""gathers all columns this table has whether defined here on in
another tables relation"""
if self.columns_cache:
return self.columns_cache
columns = {}
try:
for field in self.fields.itervalues():
for name, column in field.columns.iteritems():
if name in self.foriegn_key_columns:
continue
columns[name] = column
for name, column in self.foriegn_key_columns.iteritems():
columns[name] = column
except NoDatabaseError:
pass
return columns
def add_relations(self): # this is not a property for an optimisation
"""gathers all relations defined in this table"""
relations = {}
for field in self.fields.itervalues():
for name, relation in field.relations.iteritems():
##special rename of backref
backref = relation.sa_options.get("backref")
if backref:
new_name = backref.replace("?", self.name)
relation.sa_options["backref"] = new_name
relations[name] = relation
self.relations = relations
@property
def primary_key_columns(self):
"""gathers all primary key columns in this table"""
columns = {}
if self.primary_key:
for name, column in self.columns.iteritems():
if name in self.primary_key_list:
columns[name] = column
return columns
@property
def defined_non_primary_key_columns(self):
"""gathers all non primary key columns in this table"""
columns = {}
for name, column in self.defined_columns.iteritems():
if name not in self.primary_key_list:
columns[name] = column
return columns
def check_database(self):
"""checks if this table is part of a Database object"""
if not hasattr(self, "database"):
raise NoDatabaseError,\
"Table %s has not been assigned a database" % self.name
def _set_parent(self, database):
"""adds this table to a database object"""
if not self.table_id:
new_id = database.max_table_id + 1
self.table_id = new_id
self.kw["table_id"] = new_id
database.max_table_id = new_id
else:
database.max_table_id = max(database.max_table_id, self.table_id)
database.tables[self.name]=self
database.add_relations()
self.database = database
self.add_events()
def add_events(self):
for event in self.field_list:
if hasattr(event, "_set_actions"):
event._set_actions(self)
@property
def tables_with_relations(self):
"""returns a dictionary of all related tables and the relation object
that was defined that related them"""
self.check_database()
return self.database.tables_with_relations(self)
@property
def relation_attributes(self):
"""returns all the attributes names and accociated relation object
that appear on the sqlalchemy object"""
relation_attributes = {}
for relations in self.tables_with_relations.values():
for relation in relations:
if relation.table is self:
relation_attributes[relation.name] = relation
else:
backref = relation.sa_options["backref"]
if backref:
relation_attributes[backref] = relation
return relation_attributes
@property
def dependant_attributes(self):
"""attributes that would result in a null in the related table if
object was removed"""
dependant_attributes = {}
for table, relations in self.tables_with_relations.iteritems():
for relation in relations:
if relation.table is self and relation.original_type not in ("manytoone", "onetoone"):
dependant_attributes[relation.name] = relation
elif relation.table is not self and relation.original_type in ("manytoone", "onetoone"):
backref = relation.sa_options["backref"]
if backref:
dependant_attributes[backref] = relation
return dependant_attributes
@property
def dependant_tables(self):
dependant_tables = []
for table, relations in self.tables_with_relations.iteritems():
for relation in relations:
if relation.table is self and relation.original_type not in ("manytoone", "onetoone"):
dependant_tables.append(table[0])
elif relation.table is not self and relation.original_type in ("manytoone", "onetoone"):
dependant_tables.append(table[0])
return dependant_tables
@property
def parent_attributes(self):
parent_attributes = {}
for table, relations in self.tables_with_relations.iteritems():
for relation in relations:
if relation.table is self and relation.original_type in ("manytoone", "onetoone"):
parent_attributes[relation.name] = relation
elif relation.table is not self and relation.original_type not in ("manytoone", "onetoone"):
backref = relation.sa_options["backref"]
if backref:
parent_attributes[backref] = relation
return parent_attributes
@property
def parent_columns_attributes(self):
columns = {}
for column_name, column in self.foriegn_key_columns.iteritems():
if column.original_column <> "id":
relation = column.defined_relation
if relation.table is self and relation.original_type in ("manytoone", "onetoone"):
relation_attribute = relation.name
else:
relation_attribute = relation.sa_options["backref"]
if relation_attribute:
columns[column_name] = relation_attribute
return columns
@property
def foriegn_key_columns(self):
"""gathers the extra columns in this table that are needed as the tables
are related. i.e if this table is the many side of a one to many
relationship it will return the primary key on the "one"
side"""
if self.foriegn_key_columns_current:
return self.foriegn_key_columns_current
self.check_database()
database = self.database
columns={}
## could be made simpler
for tab, relations in self.tables_with_relations.iteritems():
table, pos = tab
for rel in relations:
if rel.foreign_key_table == self.name:
if database.tables[table].primary_key_columns:
rtable = database.tables[table]
for name, column in rtable.primary_key_columns.items():
new_col = Column(column.type,
name=name,
mandatory = rel.many_side_not_null,
defined_relation = rel,
original_column = name)
columns[name] = new_col
if rel.foreign_key_name and rel.foreign_key_name not in self.defined_columns:
columns[rel.foreign_key_name] = Column(sa.Integer,
name = rel.foreign_key_name,
mandatory = rel.many_side_not_null,
defined_relation= rel,
original_column= "id")
elif rel.foreign_key_name:
column = self.defined_columns[rel.foreign_key_name]
column.defined_relation = rel
column.original_column = "id"
columns[rel.foreign_key_name] = column
elif table+"_id" not in columns:
columns[table +'_id'] = Column(sa.Integer,
name = table +'_id',
mandatory = rel.many_side_not_null,
defined_relation= rel,
original_column= "id")
rel.foreign_key_name = rel.foreign_key_name or table + '_id'
else:
columns[table +'_id2'] = Column(sa.Integer,
name = table +'_id2',
mandatory = rel.many_side_not_null,
defined_relation= rel,
original_column= "id")
rel.foreign_key_name = rel.foreign_key_name or table + '_id2'
self.foriegn_key_columns_current = columns
return columns
@property
def foreign_key_constraints(self):
"""gathers a dictionary of all the foreign key columns for use in defining
foreign key constraints in createing sqlalchemy tables"""
fk_constraints = {}
self.check_database()
## could be made simpler
for tab, relations in self.tables_with_relations.iteritems():
table, pos = tab
## not allowed self joining constraints
if table == self.name:
continue
for rel in relations:
other_table_columns=[]
this_table_columns=[]
for name, column in self.foriegn_key_columns.iteritems():
if column.defined_relation == rel and column.original_column == "id":
other_table_columns.append("%s.%s"%\
(table, column.original_column))
this_table_columns.append(name)
if other_table_columns:
fk_constraints[rel.foreign_key_constraint_name] = [this_table_columns,
other_table_columns]
return fk_constraints
def make_sa_table(self):
"""makes a sqlalchemy table object and stores it in the
attribute sa_table"""
#make sure table is not already made
if self.sa_table:
return
self.check_database()
if not self.database.metadata:
raise NoMetadataError("table not assigned a metadata")
sa_table = sa.Table(self.name, self.database.metadata)
if self.database.engine.name == 'postgres':
statement = text("coalesce((select max(id) from (select * from public.%s) inner_table),0) + 1" % self.name)
sa_table.append_column(sa.Column("id", sa.Integer, primary_key = True, server_default = statement))
else:
sa_table.append_column(sa.Column("id", sa.Integer, primary_key = True))
for name, column in self.foriegn_key_columns.iteritems():
sa_options = column.sa_options
sa_table.append_column(sa.Column(name, column.type, **sa_options))
defined_columns = self.defined_columns
for column in self.defined_columns_order:
if column in self.foriegn_key_columns:
continue
name = column
column = defined_columns[column]
sa_options = column.sa_options
sa_table.append_column(sa.Column(name, column.type, **sa_options))
if self.foreign_key_constraints:
for name, con in self.foreign_key_constraints.iteritems():
fk_const = sa.ForeignKeyConstraint(con[0], con[1], name = name)
sa_table.append_constraint(fk_const)
for name, index in self.indexes.iteritems():
ind = [sa_table.columns[col.strip()] for col in index.fields.split(",")]
if index.type == "unique":
sa.Index(name, *ind, unique = True)
else:
sa.Index(name, *ind)
for name, constr in self.constraints.iteritems():
con = [col.strip() for col in constr.fields.split(",")]
sa_table.append_constraint(sa.UniqueConstraint(*con, name = name))
self.sa_table = sa_table
def make_sa_class(self):
"""makes a class to be mapped and stores it in the attribute sa_class"""
#make sure class is not already made
if self.sa_class:
return
table = self
class SaClass(object):
def __init__(self):
self._table = table
self._validated = False
self._version_changed = False
self._new = True
@sa.orm.reconstructor
def init_onload(self):
self._table = table
self._validated = True
self._version_changed = False
self._new = False
def __repr__(self):
return "_table: %s, id: %s, %s" % (self._table.name, self.id, ", ".join(["%s: %s" %
(field, getattr(self, field)) for
field in self._table.columns])
)
SaClass.__name__ = self.name.encode("ascii")
self.sa_class = SaClass
def sa_mapper(self):
"""runs sqlalchemy mapper to map the sa_table to sa_class and stores
the mapper object in the 'mapper' attribute"""
#make sure mapping has not been done
if self.mapper:
return
properties ={}
for col_name, column in self.columns.iteritems():
if column.name == "_version":
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = VersionChange())
elif column.parent and column.parent.type == "Password":
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertPassword())
elif column.type == sa.DateTime:
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertDate())
##FIXME do we want to keep this as date?
elif column.type == sa.Date:
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertDate())
elif column.type == sa.Boolean:
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertBoolean())
elif column.type == sa.Integer:
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertInteger())
elif column.type == sa.Unicode or isinstance(column.type, sa.Unicode):
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ConvertUnicode())
else:
properties[col_name] = column_property(getattr(self.sa_table.c,col_name),
extension = ChangedAttributes())
for relation in self.relations.itervalues():
sa_options = relation.sa_options
other_rtable = self.database.tables[relation.other]
other_table = other_rtable.sa_table
other_class = self.database.tables[relation.other].sa_class
if "backref" not in sa_options:
sa_options["backref"] = "_%s"% self.name
if sa_options["backref"]:
new_backref = sa_options["backref"].replace("?", self.name)
sa_options["backref"] = new_backref
##copy as if update_sa is run will try to add backref
sa_options = sa_options.copy()
if relation.original_type == "onetoone":
backref_string = sa_options["backref"]
if backref_string:
sa_options["backref"] = backref(backref_string, uselist = False)
if sa_options["backref"] is False:
sa_options.pop("backref")
joined_columns = []
foriegn_key_columns = []
for name, column in self.foriegn_key_columns.iteritems():
if relation == column.defined_relation and column.original_column == "id":
joined_columns.append([name, column.original_column])
foriegn_key_columns.append(getattr(self.sa_table.c, name))
if not joined_columns:
for name, col in other_rtable.foriegn_key_columns.iteritems():
if relation == col.defined_relation and col.original_column == "id":
joined_columns.append([col.original_column, name])
foriegn_key_columns.append(getattr(other_table.c, name))
join_conditions = []
for col1, col2 in joined_columns:
join_conditions.append(getattr(self.sa_table.c, col1) ==\
getattr(other_table.c, col2))
sa_options["primaryjoin"] = sa.sql.and_(*join_conditions)
sa_options["foreign_keys"] = foriegn_key_columns
sa_options["cascade"] = "none"
properties[relation.name] = sa.orm.relation(other_class,
**sa_options)
mapper_kw = dict(properties = properties)
if self.version:
mapper_kw["version_id_col"] = self.sa_table.c._version
self.mapper = mapper(self.sa_class,
self.sa_table,
**mapper_kw)
def make_paths(self):
if not self.paths:
self.paths = get_paths(self.database.graph, self.name)
if not self.local_tables:
local_tables, one_to_many_tables = make_local_tables(self.paths)
self.local_tables = local_tables
self.one_to_many_tables = one_to_many_tables
self.table_path_list = create_table_path_list(self.paths)
self.table_path = create_table_path(self.table_path_list, self.name)
def get_edge(self, table_name):
if ">" in table_name:
path = tuple(table_name.split(">")[:-1])
return self.paths[path]
return self.table_path[table_name]
def get_path(self, table_name):
return self.get_edge(table_name).path
def get_table_from_field(self, field_name):
return ".".join(field_name.split(".")[:-1])
def get_edge_from_field(self, field_name):
if field_name.count("."):
table = self.get_table_from_field(field_name)
return self.get_edge(table)
def get_path_from_field(self, field_name):
return self.get_edge_from_field(field_name).path
def validation_from_field_types(self, column):
formencode_all = All()
validators = formencode_all.validators
col_type = column.type
mand = not column.sa_options.get("nullable", True)
if column.defined_relation:
mand = False
val = formencode.validators
if column.validate == False:
pass
elif col_type is sa.Unicode or isinstance(col_type, sa.Unicode):
if hasattr(col_type, 'length'):
if col_type.length:
max = col_type.length
else:
max = 100
else:
max = 100
not_empty_string = column.mandatory
validators.append(UnicodeString(max = max,
not_empty = mand,
not_empty_string = not_empty_string))
elif col_type is sa.Integer or isinstance(col_type, sa.Integer):
validators.append(val.Int(not_empty = mand))
elif col_type is sa.Numeric or isinstance(col_type, sa.Numeric):
validators.append(val.Number(not_empty = mand))
elif col_type is sa.DateTime or isinstance(col_type, sa.DateTime):
validators.append(val.DateValidator(not_empty = mand))
elif col_type is sa.Boolean or isinstance(col_type, sa.Boolean):
validators.append(val.Bool(not_empty = mand))
elif col_type is sa.Binary or isinstance(col_type, sa.Binary):
validators.append(val.FancyValidator(not_empty = mand))
return formencode_all
def make_schema_dict(self):
schema_dict = {}
chained_validators = []
# gets from column definition
for column in self.columns.itervalues():
schema_dict[column.name] = self.validation_from_field_types(column)
if column.validation:
if column.validation.startswith("__"):
validator = validators.Regex(column.validation[2:])
else:
validator = getattr(validators, column.validation)()
schema_dict[column.name].validators.append(validator)
# gets from field definition
for field in self.fields.itervalues():
if hasattr(field, "validation"):
for name, validation in field.validation.iteritems():
schema_dict[name].validators.append(validation)
# chained validaors
if hasattr(field, "chained_validator"):
if isinstance(field.chained_validator, list):
chained_validators.extend(field.chained_validator)
else:
chained_validators.append(field.chained_validator)
# Non nullable foriegn keys are validated on the
# relationship attribute for relationships with attributes
for column in self.foriegn_key_columns.itervalues():
mand = not column.sa_options.get("nullable", True)
relation = column.defined_relation
table = relation.table
if relation and mand and not table.name.startswith("_core_"):
if relation.table is self:
attribute = relation.name
else:
attribute = relation.sa_options["backref"]
if attribute:
validator = validators.FancyValidator()
chained_validators.append(RequireIfMissing(column.name, missing = attribute))
schema_dict[attribute] = validator
else:
validation = validators.FancyValidator(not_empty = True)
schema_dict[column.name].validators.append(validation)
# many side mandatory validators
for tab, relations in self.tables_with_relations.iteritems():
for rel in relations:
if not rel.many_side_mandatory:
continue
table, pos = tab
if rel.original_type in ("onetomany", "onetooneother") and pos == "here":
schema_dict[rel.name] = validators.FancyValidator(not_empty = True)
if rel.original_type in ("manytoone", "onetoone") and pos == "other":
backref = rel.sa_options["backref"]
if backref:
validator = validators.FancyValidator(not_empty = True)
schema_dict[backref] = validator
if chained_validators:
schema_dict["chained_validators"] = chained_validators
self.schema_dict = schema_dict
@property
def schema_info(self):
return validation_rules(self.schema_dict)
def validate(self, instance, session):
"""this validates an instance of sa_class with the schema defined
by this tables Field objects"""
validation_dict = {}
for name in self.schema_dict.iterkeys():
##TODO should this be here
if name == "chained_validators":
continue
validation_dict[name] = getattr(instance, name)
instance._session = session
if not self.validated:
return {}
state = Holder(database = self.database,
table = self.name)
return validator(validation_dict, self.schema_dict, state)
def __init__(self, src=None):
super().__init__()
self._data = OrderedDict()
if src:
self.update(src)
def generate_transitions(topology, policy):
# switch_transition: dictionary of "boolean formula" : "transition".
# This will become the case statement in the NuSMV file for the
# switch transition.
switch_transition = OrderedDict()
# header_transitions:
# dictionary of "header" : {"boolean formula" : "transition"}
# This will define the transition relation on the headers in the NuSMV
# file. Headers with no transitions will become FROZENVARS.
header_transitions = {}
# For each switch configuration:
# 1. add the fowarding rules to the switch_transition dictionary as
# "string representation of boolean condition" : "new switch".
# 2. add the modification rules to header_transitions.
for switch, switch_config in policy:
assert (switch in topology)
# Ignore rules for hosts
if switch in topology.hosts():
continue
for rule in switch_config:
# Convert patterns to boolean formula
bform = ' & '.join([
'%s = %s' % (field, int_to_nusmvbitfield(field, val))
for field, val in rule.pattern.to_dict().iteritems()
if val is not None
])
if bform == '':
bform = 'switch = %s' % switch
else:
bform += ' & switch = %s' % switch
for action, vals in [unpack_action(a) for a in rule.actions]:
if action == 'modify':
# Generate transition rule
if vals[0] not in header_transitions:
header_transitions[vals[0]] = OrderedDict()
header_transitions[vals[0]][bform] = int_to_nusmvbitfield(
vals[0], vals[1])
elif action == 'forward':
if vals == ['OFPP_FLOOD']:
# TODO: implement flood
print "print_smv: flood not yet implemented."
assert (False)
elif vals == ['OFPP_CONTROLLER']:
switch_transition[bform] = 'CONTROLLER'
else:
# Find the next switch/port
out_port = vals[0]
next_switch = [
sw for sw, p in topology.node[switch]
['ports'].iteritems() if p == out_port
]
assert (len(next_switch) == 1)
next_switch = next_switch[0]
new_in_port = topology.node[next_switch]['ports'][
switch]
new_nusmv_in_port = int_to_nusmvbitfield(
IN_PORT, new_in_port)
# Generate switch transition
if next_switch in topology.hosts():
switch_transition[bform] = 'WORLD'
else:
switch_transition[bform] = next_switch
# Generate in_port transition
if IN_PORT not in header_transitions:
header_transitions[IN_PORT] = OrderedDict()
header_transitions[IN_PORT][bform] = new_nusmv_in_port
else:
# TODO: better error handling
assert (False)
# Ensure all switch transition cases are covered
for sw in topology:
default = 'switch = %s' % sw
if default not in switch_transition:
switch_transition[default] = 'DROP'
for sw in special_switches:
switch_transition['switch = %s' % sw] = sw
# Ensure all header transition cases are covered
for header in header_transitions:
header_transitions[header]['TRUE'] = header
return switch_transition, header_transitions
class ACLDB(Registry):
_class_id = 'ACLDB'
_singleton = True
def __reginit__(self):
self.rules = OrderedDict()
self.load_rules()
def load_rules(self):
pass
def save_rules(self):
pass
def reload_rules(self):
self.rules = OrderedDict()
self.load_rules()
def add_rule(self, acl, rule=None, index=None):
if rule is None:
rule = ACLRule(acl, 'False')
elif not isinstance(rule, ACLRule):
rule = ACLRule(acl, str(rule))
if not self.rules.get(acl, None):
self.rules[acl] = ACList()
if index is None:
self.rules[acl].append(rule)
return len(self.rules[acl]) - 1
else:
self.rules[acl].insert(index, rule)
return index
def set_rule(self, acl, rule, index):
if not self.rules.get(acl, None):
return False
try:
self.rules[acl][index].rule = str(rule)
return True
except IndexError:
return False
def del_rule(self, acl, index):
if not self.rules.get(acl, None):
return False
if index is None:
while len(self.rules[acl]):
self.del_rule(acl, 0)
return True
else:
try:
del self.rules[acl][index]
except IndexError:
return False
if not len(self.rules[acl]):
del self.rules[acl]
return True
def list_rules(self, name=None):
aclrules = ACList()
for rulename in self.rules.keys():
for rule in self.rules[rulename]:
if name in (None, rulename):
aclrules.append(rule)
return aclrules
def get_namespace(self, **namespaces):
namespace = dict(self.rules)
namespace['proxy'] = ProxyNamespace()
for ns in namespaces:
if not namespace.has_key(ns):
namespace[ns] = ACLTags()
namespace[ns].update(namespaces[ns])
return namespace
def eval(self, expression, **namespaces):
"""
Evaluate an ACL expression in the namespaces context.
"""
aclparser = ACLRuleParser(
namespace=self.get_namespace(**namespaces))
return aclparser.eval(expression)
def check(self, acl, **namespaces):
"""
Check an ACL rule in the namespaces context.
"""
try:
namespace = self.get_namespace(**namespaces)
result = None
match = ''
if isinstance(acl, ACLRule):
match = repr(acl.rule)
result = acl.eval(namespace)
else:
if self.rules.has_key(acl):
for rule in self.rules[acl]:
match = repr(rule.rule)
if rule.eval(namespace):
result = True
break
else:
result = False
if result is None:
result = False
log.info('ACL %s not found' % str(acl))
else:
log.info('ACL %s %s %s' % (acl, result, match))
pass
return result
except:
log.exception('Error while checking ACL %s' % str(acl))
raise
class Database(object):
def __init__(self, *tables, **kw):
log.info("initialising database")
self.status = "updating"
self.kw = kw
self.metadata = None
self.connection_string = kw.get("connection_string", None)
if self.connection_string:
self.engine = create_engine(self.connection_string)
self.metadata = MetaData()
self.metadata.bind = self.engine
self._Session = sessionmaker(bind=self.engine, autoflush = False)
self.Session = sessionwrapper.SessionClass(self._Session, self)
self.logging_tables = kw.get("logging_tables", None)
self.quiet = kw.get("quiet", None)
self.application = kw.get("application", None)
if self.application:
self.set_application(self.application)
self.max_table_id = 0
self.max_event_id = 0
self.persisted = False
self.graph = None
self.relations = []
self.tables = OrderedDict()
self.search_actions = {}
self.search_names = {}
self.search_ids = {}
for table in tables:
self.add_table(table)
def set_application(self, application):
self.application = application
if not self.connection_string:
self.metadata = application.metadata
self.engine = application.engine
self._Session = application.Session
self.Session = sessionwrapper.SessionClass(self._Session, self)
if self.logging_tables is not None:
self.logging_tables = self.application.logging_tables
if self.quiet is not None:
self.quiet = self.application.quiet
self.application_folder = self.application.application_folder
self.zodb = self.application.zodb
self.zodb_tables_init()
self.application.Session = self.Session
with SchemaLock(self) as file_lock:
self.load_from_persist()
self.status = "active"
def zodb_tables_init(self):
zodb = self.application.aquire_zodb()
connection = zodb.open()
root = connection.root()
if "tables" not in root:
root["tables"] = PersistentMapping()
root["table_count"] = 0
root["event_count"] = 0
transaction.commit()
connection.close()
zodb.close()
self.application.get_zodb(True)
def __getitem__(self, item):
if isinstance(item, int):
table = self.get_table_by_id(item)
if not table:
raise IndexError("table id %s does not exist" % item)
return table
else:
return self.tables[item]
def get_table_by_id(self, id):
for table in self.tables.itervalues():
if table.table_id == id:
return table
def add_table(self, table, ignore = False, drop = False):
log.info("adding table %s" % table.name)
if table.name in self.tables.iterkeys():
if ignore:
return
elif drop:
self.drop_table(table.name)
else:
raise custom_exceptions.DuplicateTableError("already a table named %s"
% table.name)
self._add_table_no_persist(table)
def rename_table(self, table, new_name, session = None):
if isinstance(table, tables.Table):
table_to_rename = table
else:
table_to_rename = self.tables[table]
with SchemaLock(self) as file_lock:
for relations in table_to_rename.tables_with_relations.values():
for rel in relations:
if rel.other == table_to_rename.name:
field = rel.parent
field.args = [new_name] + list(field.args[1:])
table_to_rename.name = new_name
file_lock.export(uuid = True)
table_to_rename.sa_table.rename(new_name)
file_lock.export()
self.load_from_persist(True)
if table_to_rename.logged:
self.rename_table("_log_%s" % table_to_rename.name, "_log_%s" % new_name, session)
def drop_table(self, table):
with SchemaLock(self) as file_lock:
if isinstance(table, tables.Table):
table_to_drop = table
else:
table_to_drop = self.tables[table]
if table_to_drop.dependant_tables:
raise custom_exceptions.DependencyError((
"cannot delete table %s as the following tables"
" depend on it %s" % (table.name, table.dependant_tables)))
for relations in table_to_drop.tables_with_relations.itervalues():
for relation in relations:
field = relation.parent
field.table.fields.pop(field.name)
field.table.field_list.remove(field)
self.tables.pop(table_to_drop.name)
file_lock.export(uuid = True)
table_to_drop.sa_table.drop()
file_lock.export()
self.load_from_persist(True)
if table_to_drop.logged:
self.drop_table(self.tables["_log_" + table_to_drop.name])
def add_relation_table(self, table):
if "_core" not in self.tables:
raise custom_exceptions.NoTableAddError("table %s cannot be added as there is"
"no _core table in the database"
% table.name)
assert table.primary_entities
assert table.secondary_entities
table.relation = True
table.kw["relation"] = True
self.add_table(table)
relation = ForeignKey("_core_id", "_core", backref = table.name)
table._add_field_no_persist(relation)
event = Event("delete",
actions.DeleteRows("_core"))
table.add_event(event)
def add_info_table(self, table):
if "_core" not in self.tables:
raise custom_exceptions.NoTableAddError("table %s cannot be added as there is"
"no _core table in the database"
% table.name)
table.info_table = True
table.kw["info_table"] = True
self.add_table(table)
relation = ForeignKey("_core_id", "_core", backref = table.name)
table._add_field_no_persist(relation)
event = Event("delete",
actions.DeleteRows("_core"))
table.add_event(event)
def add_entity(self, table):
if "_core" not in self.tables:
raise custom_exceptions.NoTableAddError("table %s cannot be added as there is"
"no _core table in the database"
% table.name)
table.entity = True
table.kw["entity"] = True
self.add_table(table)
#add relation
relation = ForeignKey("_core_id", "_core", backref = table.name)
table._add_field_no_persist(relation)
##add title events
if table.title_field:
title_field = table.title_field
else:
title_field = "name"
event = Event("new change",
actions.CopyTextAfter("primary_entity._core_entity.title", title_field))
table.add_event(event)
if table.summary_fields:
event = Event("new change",
actions.CopyTextAfterField("primary_entity._core_entity.summary", table.summary_fields))
table.add_event(event)
event = Event("delete",
actions.DeleteRows("primary_entity._core_entity"))
table.add_event(event)
def _add_table_no_persist(self, table):
table._set_parent(self)
def persist(self):
self.status = "updating"
for table in self.tables.values():
if not self.logging_tables:
## FIXME should look at better place to set this
table.kw["logged"] = False
table.logged = False
if table.logged and "_log_%s" % table.name not in self.tables.iterkeys() :
self.add_table(self.logged_table(table))
for table in self.tables.itervalues():
table.add_foreign_key_columns()
self.update_sa(True)
with SchemaLock(self) as file_lock:
file_lock.export(uuid = True)
self.metadata.create_all(self.engine)
self.persisted = True
file_lock.export()
self.load_from_persist(True)
def get_file_path(self, uuid_name = False):
uuid = datetime.datetime.now().isoformat().\
replace(":", "-").replace(".", "-")
if uuid_name:
file_name = "generated_schema-%s.py" % uuid
else:
file_name = "generated_schema.py"
file_path = os.path.join(
self.application.application_folder,
"_schema",
file_name
)
return file_path
def code_repr_load(self):
import _schema.generated_schema as sch
sch = reload(sch)
database = sch.database
database.clear_sa()
for table in database.tables.values():
table.database = self
self.add_table(table)
table.persisted = True
self.max_table_id = database.max_table_id
self.max_event_id = database.max_event_id
self.persisted = True
def code_repr_export(self, file_path):
try:
os.remove(file_path)
os.remove(file_path+"c")
except OSError:
pass
out_file = open(file_path, "w")
output = [
"from database.database import Database",
"from database.tables import Table",
"from database.fields import *",
"from database.database import table, entity, relation",
"from database.events import Event",
"from database.actions import *",
"",
"",
"database = Database(",
"",
"",
]
for table in sorted(self.tables.values(),
key = lambda x:x.table_id):
output.append(table.code_repr() + ",")
kw_display = ""
if self.kw:
kw_list = ["%s = %s" % (i[0], repr(i[1])) for i in self.kw.items()]
kw_display = ", ".join(sorted(kw_list))
output.append(kw_display)
output.append(")")
out_file.write("\n".join(output))
out_file.close()
return file_path
def load_from_persist(self, restart = False):
self.clear_sa()
self.tables = OrderedDict()
try:
self.code_repr_load()
except ImportError:
return
self.add_relations()
self.update_sa()
self.validate_database()
def add_relations(self): #not property for optimisation
self.relations = []
for table_name, table_value in self.tables.iteritems():
## make sure fk columns are remade
table_value.foriegn_key_columns_current = None
table_value.add_relations()
for rel_name, rel_value in table_value.relations.iteritems():
self.relations.append(rel_value)
def checkrelations(self):
for relation in self.relations:
if relation.other not in self.tables.iterkeys():
raise custom_exceptions.RelationError,\
"table %s does not exits" % relation.other
def update_sa(self, reload = False):
if reload == True and self.status <> "terminated":
self.status = "updating"
if reload:
self.clear_sa()
self.checkrelations()
self.make_graph()
try:
for table in self.tables.itervalues():
table.make_paths()
table.make_sa_table()
table.make_sa_class()
for table in self.tables.itervalues():
table.sa_mapper()
sa.orm.compile_mappers()
for table in self.tables.itervalues():
for column in table.columns.iterkeys():
getattr(table.sa_class, column).impl.active_history = True
table.columns_cache = table.columns
for table in self.tables.itervalues():
table.make_schema_dict()
## put valid_info tables into info_table
for table in self.tables.itervalues():
if table.relation or table.entity:
for valid_info_table in table.valid_info_tables:
info_table = self.tables[valid_info_table]
assert info_table.info_table
info_table.valid_core_types.append(table.name)
self.collect_search_actions()
except (custom_exceptions.NoDatabaseError,\
custom_exceptions.RelationError):
pass
if reload == True and self.status <> "terminated":
self.status = "active"
def clear_sa(self):
sa.orm.clear_mappers()
if self.metadata:
self.metadata.clear()
for table in self.tables.itervalues():
table.foriegn_key_columns_current = None
table.mapper = None
table.sa_class = None
table.sa_table = None
table.paths = None
table.local_tables = None
table.one_to_many_tables = None
table.events = dict(new = [],
delete = [],
change = [])
table.schema_dict = None
table.valid_core_types = []
table.columns_cache = None
self.graph = None
self.search_actions = {}
self.search_names = {}
self.search_ids = {}
def tables_with_relations(self, table):
relations = defaultdict(list)
for n, v in table.relations.iteritems():
relations[(v.other, "here")].append(v)
for v in self.relations:
if v.other == table.name:
relations[(v.table.name, "other")].append(v)
return relations
def result_set(self, search):
return resultset.ResultSet(search)
def search(self, table_name, where = "id>0", *args, **kw):
##FIXME id>0 should be changed to none once search is sorted
"""
:param table_name: specifies the base table you will be query from (required)
:param where: either a paramatarised or normal where clause, if paramitarised
either values or params keywords have to be added. (optional first arg, if
missing will query without where)
:param tables: an optional list of onetoone or manytoone tables to be extracted
with results
:param keep_all: will keep id, _core_entity_id, modified_by and modified_on fields
:param fields: an optional explicit field list in the form 'field' for base table
and 'table.field' for other tables. Overwrites table option and keep all.
:param limit: the row limit
:param offset: the offset
:param internal: if true will not convert date, boolean and decimal fields
:param values: a list of values to replace the ? in the paramatarised queries
:param params: a dict with the keys as the replacement to inside the curly
brackets i.e key name will replace {name} in query.
:param order_by: a string in the same form as a sql order by
ie 'name desc, donkey.name, donkey.age desc' (name in base table)
"""
session = kw.pop("session", None)
if session:
external_session = True
else:
session = self.Session()
external_session = False
tables = kw.get("tables", [table_name])
fields = kw.get("fields", None)
join_tables = []
if fields:
join_tables = split_table_fields(fields, table_name).keys()
if table_name in join_tables:
join_tables.remove(table_name)
tables = None
if tables:
join_tables.extend(tables)
if table_name in tables:
join_tables.remove(table_name)
if "order_by" not in kw:
kw["order_by"] = "id"
if join_tables:
kw["extra_outer"] = join_tables
kw["distinct_many"] = False
try:
query = search.Search(self, table_name, session, where, *args, **kw)
result = resultset.ResultSet(query, **kw)
result.collect()
return result
except Exception, e:
session.rollback()
raise
finally:
def __reginit__(self):
self.rules = OrderedDict()
self.load_rules()
class ACLDB(Registry):
_class_id = 'ACLDB'
_singleton = True
def __reginit__(self):
self.rules = OrderedDict()
self.load_rules()
def load_rules(self):
pass
def save_rules(self):
pass
def reload_rules(self):
self.rules = OrderedDict()
self.load_rules()
def add_rule(self, acl, rule=None, index=None):
if rule is None:
rule = ACLRule(acl, 'False')
elif not isinstance(rule, ACLRule):
rule = ACLRule(acl, str(rule))
if not self.rules.get(acl, None):
self.rules[acl] = ACList()
if index is None:
self.rules[acl].append(rule)
return len(self.rules[acl]) - 1
else:
self.rules[acl].insert(index, rule)
return index
def set_rule(self, acl, rule, index):
if not self.rules.get(acl, None):
return False
try:
self.rules[acl][index].rule = str(rule)
return True
except IndexError:
return False
def del_rule(self, acl, index):
if not self.rules.get(acl, None):
return False
if index is None:
while len(self.rules[acl]):
self.del_rule(acl, 0)
return True
else:
try:
del self.rules[acl][index]
except IndexError:
return False
if not len(self.rules[acl]):
del self.rules[acl]
return True
def list_rules(self, name=None):
aclrules = ACList()
for rulename in self.rules.keys():
for rule in self.rules[rulename]:
if name in (None, rulename):
aclrules.append(rule)
return aclrules
def get_namespace(self, **namespaces):
namespace = dict(self.rules)
namespace['proxy'] = ProxyNamespace()
for ns in namespaces:
if not namespace.has_key(ns):
namespace[ns] = ACLTags()
namespace[ns].update(namespaces[ns])
return namespace
def eval(self, expression, **namespaces):
"""
Evaluate an ACL expression in the namespaces context.
"""
aclparser = ACLRuleParser(namespace=self.get_namespace(**namespaces))
return aclparser.eval(expression)
def check(self, acl, **namespaces):
"""
Check an ACL rule in the namespaces context.
"""
try:
namespace = self.get_namespace(**namespaces)
result = None
match = ''
if isinstance(acl, ACLRule):
match = repr(acl.rule)
result = acl.eval(namespace)
else:
if self.rules.has_key(acl):
for rule in self.rules[acl]:
match = repr(rule.rule)
if rule.eval(namespace):
result = True
break
else:
result = False
if result is None:
result = False
log.info('ACL %s not found' % str(acl))
else:
log.info('ACL %s %s %s' % (acl, result, match))
pass
return result
except:
log.exception('Error while checking ACL %s' % str(acl))
raise
def __init__(self, name, *args , **kw):
"""
name: name of the table
primary_key: a comma delimited string stating what field
should act as a primary key.
logged: Boolean stating if the table should be logged
modified_by: Boolean stating if the table should have a last
modified by
*args : All the Field objects this table has.
"""
self.all_kw = set(self.modifiable_kw + self.restricted_kw)
self.all_updatable_kw = set(self.modifiable_kw)
assert set(kw.keys()) - self.all_kw == set()
self.name = name
self.kw = kw
self.args = args
self.table_id = kw.get("table_id", None)
self.field_list = []
self.fields = OrderedDict()
self.field_order = kw.get("field_order", [])
self.current_order = 0
self.primary_key = kw.get("primary_key", None)
#persisted should be private
self.persisted = kw.get("persisted", False)
self.quiet = kw.get("quiet", False)
# table types
self.entity = kw.get("entity", False)
self.relation = kw.get("relation", False)
self.lookup = kw.get("lookup", False)
self.info_table = kw.get("info_table", False)
self.table_class = kw.get("table_class")
## for relaion tables
primary_entities = kw.get("primary_entities")
if primary_entities :
self.primary_entities = primary_entities.strip().split(",")
secondary_entities = kw.get("secondary_entities")
if secondary_entities:
self.secondary_entities = secondary_entities.strip().split(",")
## for entity or relation tables
valid_info_tables = kw.get("valid_info_tables")
if valid_info_tables:
self.valid_info_tables = valid_info_tables.split()
else:
self.valid_info_tables = []
## for info tables to be populated
self.valid_core_types = []
self.max_field_id = 0
self.logged = kw.get("logged", True)
self.validated = kw.get("validated", True)
self.modified_date = kw.get("modified_date", True)
self.modified_by = kw.get("modified_by", True)
self.version = kw.get("version", True)
self.table_type = kw.get("table_type", "user")
self.title_field = kw.get("title_field", None)
self.description_field = kw.get("description_field", None)
self.summary_fields = kw.get("summary_fields", None)
self.summary = kw.get("summary", None)
self.default_node = kw.get("default_node", None)
self.primary_key_list = []
if self.primary_key:
self.primary_key_list = self.primary_key.split(",")
self.events = dict(new = [],
delete = [],
change = [])
for field in args:
field._set_parent(self)
if "_version" not in self.fields and self.version:
self.add_field(Integer("_version"))
if "_modified_date" not in self.fields and self.modified_date:
self.add_field(Modified("_modified_date"))
if "_modified_by" not in self.fields and self.modified_by:
self.add_field(ModifiedBySession("_modified_by"))
self.foriegn_key_columns_current = None
#sqlalchemy objects
self.sa_table = None
self.sa_class = None
self.mapper = None
self.paths = None
self.local_tables = None
self.one_to_many_tables = None
self.table_path_list = None
self.table_path = None
self.schema_dict = None
self.schema = None
self.columns_cache = None
class OrderedSet:
def __init__(self, src=None):
super().__init__()
self._data = OrderedDict()
if src:
self.update(src)
def update(self, *sources):
for src in sources:
for item in src:
self.add(item)
return self
def difference_update(self, src):
for item in src:
self.discard(item)
return self
def add(self, item):
if item in self:
return False
self._data[item] = None
return True
def __contains__(self, item):
return item in self._data
def discard(self, item):
if item in self:
return False
del self._data[item]
return True
def remove(self, item):
if not self.discard(item):
raise KeyError
def __iter__(self):
yield from self._data
def clear(self):
if not self:
return False
self._data.clear()
return True
def __bool__(self):
return bool(self._data)
def __repr__(self):
return f"{{{', '.join(repr(e) for e in self)}}}"
def __len__(self):
return len(self._data)
def __getitem__(self, index):
res = list(self)
if index == slice(None):
return res
return res[index]
def reload_rules(self):
self.rules = OrderedDict()
self.load_rules()