Revised db.select to be completely general
This commit is contained in:
Doug Blank
parent
dc698782b9
commit
f9930c1bcf
@ -204,66 +204,105 @@ class BasicPrimaryObject(TableObject, PrivacyBase, TagBase):
|
|||||||
|
|
||||||
def _follow_field_path(self, chain, db=None, ignore_errors=False):
|
def _follow_field_path(self, chain, db=None, ignore_errors=False):
|
||||||
"""
|
"""
|
||||||
Follow a list of items. Return endpoint.
|
Follow a list of items. Return endpoint(s) only.
|
||||||
With the db argument, can do joins across tables.
|
With the db argument, can do joins across tables.
|
||||||
self - current object
|
self - current object
|
||||||
|
returns - None, endpoint, of recursive list of endpoints
|
||||||
"""
|
"""
|
||||||
from .handle import HandleClass
|
from .handle import HandleClass
|
||||||
|
# start with [self, self, chain, path_to=[]]
|
||||||
|
# results = []
|
||||||
|
# expand when you reach multiple answers [obj, chain_left, []]
|
||||||
|
# if you get to an endpoint, put results
|
||||||
|
# go until nothing left to expand
|
||||||
current = self
|
current = self
|
||||||
path_to = []
|
todo = [(self, current, chain, [])]
|
||||||
parent = self
|
results = []
|
||||||
for part in chain:
|
while todo:
|
||||||
path_to.append(part)
|
parent, current, chain, path_to = todo.pop()
|
||||||
if hasattr(current, part): # attribute
|
keep_going = True
|
||||||
current = getattr(current, part)
|
p = 0
|
||||||
elif part.isdigit(): # index into list
|
while p < len(chain) and keep_going:
|
||||||
if int(part) < len(current):
|
part = chain[p]
|
||||||
current = current[int(part)]
|
if hasattr(current, part): # attribute
|
||||||
continue
|
current = getattr(current, part)
|
||||||
elif ignore_errors:
|
path_to.append(part)
|
||||||
return
|
# need to consider current+part if current is list:
|
||||||
else:
|
elif isinstance(current, (list, tuple)):
|
||||||
raise Exception("Can't get position %s of %s" % (part, current))
|
if part.isdigit():
|
||||||
elif isinstance(current, (list, tuple)):
|
# followed by index, so continue here
|
||||||
current = [getattr(attr, part) for attr in current]
|
current = current[int(part)]
|
||||||
else: # part not found on this self
|
path_to.append(part)
|
||||||
# current is a handle
|
else: # else branch! in middle, split paths
|
||||||
# part is something on joined object
|
for i in range(len(current)):
|
||||||
ptype = parent.__class__.get_field_type(".".join(path_to[:-1]))
|
todo.append([self, current, [str(i)] + chain[p:], path_to])
|
||||||
if isinstance(ptype, HandleClass):
|
current = None
|
||||||
if db:
|
keep_going = False
|
||||||
# start over here:
|
else: # part not found on this self
|
||||||
try:
|
# current is a handle
|
||||||
parent = ptype.join(db, current)
|
# part is something on joined object
|
||||||
current = getattr(parent, part)
|
if parent:
|
||||||
path_to = []
|
ptype = parent.__class__.get_field_type(".".join(path_to))
|
||||||
continue
|
if isinstance(ptype, HandleClass):
|
||||||
except:
|
if db:
|
||||||
if ignore_errors:
|
# start over here:
|
||||||
return
|
obj = ptype.join(db, current)
|
||||||
|
if part == "self":
|
||||||
|
current = obj
|
||||||
|
elif obj:
|
||||||
|
current = getattr(obj, part)
|
||||||
|
if current:
|
||||||
|
path_to = []
|
||||||
|
todo.append([obj, current, chain[p + 1:], path_to])
|
||||||
|
current = None
|
||||||
|
keep_going = False
|
||||||
else:
|
else:
|
||||||
raise
|
raise Exception("Can't join without database")
|
||||||
else:
|
elif part == "self":
|
||||||
raise Exception("Can't join without database")
|
pass
|
||||||
if ignore_errors:
|
elif ignore_errors:
|
||||||
return
|
pass
|
||||||
else:
|
else:
|
||||||
raise Exception("%s is not a valid field of %s; use %s" %
|
raise Exception("%s is not a valid field of %s; use %s" %
|
||||||
(part, current, dir(current)))
|
(part, current, dir(current)))
|
||||||
return current
|
current = None
|
||||||
|
keep_going = False
|
||||||
|
p += 1
|
||||||
|
if keep_going:
|
||||||
|
results.append(current)
|
||||||
|
if len(results) == 1:
|
||||||
|
return results[0]
|
||||||
|
elif len(results) == 0:
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
return results
|
||||||
|
|
||||||
def set_field(self, field, value, db=None, ignore_errors=False):
|
def set_field(self, field, value, db=None, ignore_errors=False):
|
||||||
"""
|
"""
|
||||||
Set the value of a basic field (str, int, float, or bool).
|
Set the value of a basic field (str, int, float, or bool).
|
||||||
value can be a string or actual value.
|
value can be a string or actual value.
|
||||||
|
Returns number of items changed.
|
||||||
"""
|
"""
|
||||||
field = self.__class__.get_field_alias(field)
|
field = self.__class__.get_field_alias(field)
|
||||||
chain = field.split(".")
|
chain = field.split(".")
|
||||||
path = self._follow_field_path(chain[:-1], db, ignore_errors)
|
path = self._follow_field_path(chain[:-1] + ["self"], db, ignore_errors)
|
||||||
ftype = self.get_field_type(field)
|
ftype = self.get_field_type(field)
|
||||||
# ftype is str, bool, float, or int
|
# ftype is str, bool, float, or int
|
||||||
value = (value in ['True', True]) if ftype is bool else value
|
value = (value in ['True', True]) if ftype is bool else value
|
||||||
setattr(path, chain[-1], ftype(value))
|
return self._set_fields(path, chain[-1], value, ftype)
|
||||||
|
|
||||||
|
def _set_fields(self, path, attr, value, ftype):
|
||||||
|
"""
|
||||||
|
Helper function to handle recursive lists of items.
|
||||||
|
"""
|
||||||
|
if isinstance(path, (list, tuple)):
|
||||||
|
count = 0
|
||||||
|
for item in path:
|
||||||
|
count += self._set_fields(item, attr, value, ftype)
|
||||||
|
else:
|
||||||
|
setattr(path, attr, ftype(value))
|
||||||
|
count = 1
|
||||||
|
return count
|
||||||
|
|
||||||
def set_gramps_id(self, gramps_id):
|
def set_gramps_id(self, gramps_id):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user