def _bind_type(self, obj):
"""
If object has a type (publication_type, organization_type, event_type etc.)
this method assigns type from the database enum to this object. If the type
doesn't exist, method returns False. If everything went OK, returns True.
If object doesn't have a type, raises DatabaseError.
The object itself has to be inserted into database before binding type!
(it fact, it has to contain attribute ID). If not, method raises Exception.
"""
if obj['id'] is None:
raise DatabaseError("Object has to contain attribute ID. Insert it into db first.")
if 'type' in obj:
type_name = obj['type']['type']
q = FluentSQLQuery()
q.select("id").from_table(obj['type']._table_name).where("type=", type_name)
q()
res = q.fetch_one()
if res is None:
return False
type_id = res['id']
q.cleanup()
q.update(obj._table_name, {"type_id": type_id}).where("id=", obj['id'])
q()
return q.count() == 1
else:
raise RRSDatabaseEntityError("Entity doesn't have a type to bind to.")
def _bind_type(self, obj):
"""
If object has a type (publication_type, organization_type, event_type etc.)
this method assigns type from the database enum to this object. If the type
doesn't exist, method returns False. If everything went OK, returns True.
If object doesn't have a type, raises DatabaseError.
The object itself has to be inserted into database before binding type!
(it fact, it has to contain attribute ID). If not, method raises Exception.
"""
if obj["id"] is None:
raise DatabaseError("Object has to contain attribute ID. Insert it into db first.")
if "type" in obj:
type_name = obj["type"]["type"]
q = FluentSQLQuery()
q.select("id").from_table(obj["type"]._table_name).where("type=", type_name)
q()
res = q.fetch_one()
if res is None:
return False
type_id = res["id"]
q.cleanup()
q.update(obj._table_name, {"type_id": type_id}).where("id=", obj["id"])
q()
return q.count() == 1
else:
raise RRSDatabaseEntityError("Entity doesn't have a type to bind to.")
def _bind_entity_to_name(self, namedentity, source_module):
"""
This method creates connection between entity and it's name, which is stored
in other database table. These tables are:
- person vs person_name
- organization vs organization_name
- event vs event_name
@returns ID of the row in the name-table.
"""
ACRONYM = 'acronym' # to be easily changed to abbreviation or whatever needed..
TITLE = 'title' # will be name? or what?
if not isinstance(namedentity, _RRSDatabaseEntity):
raise TypeError("Named object has to be instance of subclass of _RRSDatabaseEntity")
if not 'id' in namedentity:
raise DatabaseError("Named object has to contain ID!")
q = FluentSQLQuery()
if namedentity._table_name == "person":
# act like person and handle person_name
# this is slightly different because there is N:N relationship
# create new person name object
pname = RRSPerson_name()
for attr in ('first_name', 'middle_name', 'last_name', 'full_name'):
if attr in namedentity:
pname[attr] = namedentity[attr]
# create relationship object
rel_obj = RRSRelationshipPersonPerson_name()
rel_obj.set_entity(pname)
namedentity['person_name'] = rel_obj
# look for this name in database
if self.lookup(pname):
# it is in db yet, just check if rel exists
q.select("person_id").from_table(("j__person__person_name"))
q.where("person_id=", namedentity['id']).and_("person_name_id=", pname['id'])
q()
if not q.count():
# if the relationship doesn't exist, create new one
self._rrsdb.relationship("person_name", rel_obj)
elif q.count() > 1:
self.logger.warning("There are more than one relationship "\
"entries in table 'j__person__person_name"\
" between person.id=%s and person_name.id=%s" \
% (namedentity['id'], pname['id']))
else:
# insert new person_name and create the relationship
self._rrsdb.insert(pname, self.module)
self._rrsdb.relationship("person_name", rel_obj)
# get the reference out of which is this name extracted and assign
# the person name to the reference (j__person_name__reference)
try:
refe = namedentity['publication'][0].get_entities()[0]['reference_reference'][0].get_entities()[0]
except (KeyError, TypeError, IndexError):
pass
else:
pname_ref_rel = RRSRelationshipPerson_nameReference()
pname_ref_rel.set_entity(refe)
pname['reference'] = pname_ref_rel
try:
self._rrsdb.relationship('reference', pname_ref_rel)
except DatabaseError:
self._queue.wait(WQEntry(self._rrsdb.relationship, ('reference', pname_ref_rel)))
elif namedentity._table_name in ("event", "organization"):
if TITLE not in namedentity:
# this violates constraint... raise exception?? Or return false?
return False
name_tbl = "%s_name" % namedentity._table_name
# if there in the database is no title like this, insert it
q.select(("id", "%s_id" % namedentity._table_name, ACRONYM, TITLE)).from_table(name_tbl)
q.where("%s=" % TITLE, namedentity[TITLE])
if ACRONYM in namedentity:
q.or_("%s=" % ACRONYM, namedentity[ACRONYM])
q()
if q.count():
# check the parent id if it matches
for row in q.fetch_all():
if namedentity['id'] == row[1]:
# if it matched on acronym, check the titles if they are the same
if row[TITLE] != namedentity[TITLE]:
# if not, check the rest and maybe add new row into table
continue
# add the missing acronym if needed
if row[ACRONYM] is None and namedentity[ACRONYM] is not None:
# update the row
q.cleanup()
q.update(name_tbl, {ACRONYM: namedentity[ACRONYM]})
q.where("id=", row['id'])
q()
return row['id']
# if nothing matched, insert new name
name_obj = self._table_to_class_map[name_tbl]()
for attr in (TITLE, ACRONYM):
if attr in namedentity:
name_obj[attr] = namedentity[attr]
if name_obj.empty():
return False
name_obj[namedentity._table_name] = namedentity
self._rrsdb.insert(name_obj, source_module)
return name_obj['id']
else:
raise RRSDatabaseEntityError("%s is not a named entity." % type(namedentity))
self._db.refresh()
def _bind_entity_to_name(self, namedentity, source_module):
"""
This method creates connection between entity and it's name, which is stored
in other database table. These tables are:
- person vs person_name
- organization vs organization_name
- event vs event_name
@returns ID of the row in the name-table.
"""
ACRONYM = "acronym" # to be easily changed to abbreviation or whatever needed..
TITLE = "title" # will be name? or what?
if not isinstance(namedentity, _RRSDatabaseEntity):
raise TypeError("Named object has to be instance of subclass of _RRSDatabaseEntity")
if not "id" in namedentity:
raise DatabaseError("Named object has to contain ID!")
q = FluentSQLQuery()
if namedentity._table_name == "person":
# act like person and handle person_name
# this is slightly different because there is N:N relationship
# create new person name object
pname = RRSPerson_name()
for attr in ("first_name", "middle_name", "last_name", "full_name"):
if attr in namedentity:
pname[attr] = namedentity[attr]
# create relationship object
rel_obj = RRSRelationshipPersonPerson_name()
rel_obj.set_entity(pname)
namedentity["person_name"] = rel_obj
# look for this name in database
if self.lookup(pname):
# it is in db yet, just check if rel exists
q.select("person_id").from_table(("j__person__person_name"))
q.where("person_id=", namedentity["id"]).and_("person_name_id=", pname["id"])
q()
if not q.count():
# if the relationship doesn't exist, create new one
self._rrsdb.relationship("person_name", rel_obj)
elif q.count() > 1:
self.logger.warning(
"There are more than one relationship "
"entries in table 'j__person__person_name"
" between person.id=%s and person_name.id=%s" % (namedentity["id"], pname["id"])
)
else:
# insert new person_name and create the relationship
self._rrsdb.insert(pname, self.module)
self._rrsdb.relationship("person_name", rel_obj)
# get the reference out of which is this name extracted and assign
# the person name to the reference (j__person_name__reference)
try:
refe = namedentity["publication"][0].get_entities()[0]["reference_reference"][0].get_entities()[0]
except (KeyError, TypeError, IndexError):
pass
else:
pname_ref_rel = RRSRelationshipPerson_nameReference()
pname_ref_rel.set_entity(refe)
pname["reference"] = pname_ref_rel
try:
self._rrsdb.relationship("reference", pname_ref_rel)
except DatabaseError:
self._queue.wait(WQEntry(self._rrsdb.relationship, ("reference", pname_ref_rel)))
elif namedentity._table_name in ("event", "organization"):
if TITLE not in namedentity:
# this violates constraint... raise exception?? Or return false?
return False
name_tbl = "%s_name" % namedentity._table_name
# if there in the database is no title like this, insert it
q.select(("id", "%s_id" % namedentity._table_name, ACRONYM, TITLE)).from_table(name_tbl)
q.where("%s=" % TITLE, namedentity[TITLE])
if ACRONYM in namedentity:
q.or_("%s=" % ACRONYM, namedentity[ACRONYM])
q()
if q.count():
# check the parent id if it matches
for row in q.fetch_all():
if namedentity["id"] == row[1]:
# if it matched on acronym, check the titles if they are the same
if row[TITLE] != namedentity[TITLE]:
# if not, check the rest and maybe add new row into table
continue
# add the missing acronym if needed
if row[ACRONYM] is None and namedentity[ACRONYM] is not None:
# update the row
q.cleanup()
q.update(name_tbl, {ACRONYM: namedentity[ACRONYM]})
q.where("id=", row["id"])
q()
return row["id"]
# if nothing matched, insert new name
name_obj = self._table_to_class_map[name_tbl]()
for attr in (TITLE, ACRONYM):
if attr in namedentity:
name_obj[attr] = namedentity[attr]
if name_obj.empty():
return False
name_obj[namedentity._table_name] = namedentity
self._rrsdb.insert(name_obj, source_module)
return name_obj["id"]
else:
raise RRSDatabaseEntityError("%s is not a named entity." % type(namedentity))
self._db.refresh()
