def add_file(new_file, return_updated = False):
new_file = Files( parent = ndb.Key("FileTable", "*notitle*"),
file_name = new_file.file_name,
owner_key = new_file.owner_key,
friendly_name = new_file.friendly_name,
prev_file_key = new_file.prev_file_key,
axis_a = new_file.axis_a,
axis_b = new_file.axis_b)
new_key = new_file.put();
if return_updated:
new_file.key = new_key
else:
return new_key
def get_file_by_owner_key(owner_key): if isinstance(owner_key,ndb.Key): query = Files.query(Files.owner_key == owner_key) iterator = query.iter() return iterator else: return None
def get_file_by_name(file_name): query = Files.query(Files.file_name == file_name) file = query.get() if file is None: return None else: return FileInfo( file_name = file.file_name, owner_key = file.owner_key, friendly_name = file.friendly_name, key = file.key, prev_file_key = file.prev_file_key, axis_a = file.axis_a, axis_b = file.axis_b)
