def main():
parser = ArgumentParser(description='Tag files')
parser.add_argument('file', help='File to tag', type=lambda file: is_file_valid(parser, file))
parser.add_argument('-t', '--tags', help='Add tags to file')
parser.add_argument('-e', '--edit', help='Edit tags for file', action='store_true')
parser.add_argument('--list', help='List all tags', action='store_true')
doc = Document(parser.parse_args().file)
if parser.parse_args().edit:
if parser.parse_args().tags:
doc.edit(parser.parse_args().tags)
else:
print('Add -t to edit tags')
else:
doc.add(tags=parser.parse_args().tags)
if parser.parse_args().list:
tags = Tags()
print(tags.get('database'))
doc.save_to_db()
def add_document(self):
for i in range(len(self.fileinfo)):
new_doc = Document(self.fileinfo[i])
new_doc.add(self.tag_input.text())
new_doc.save_to_db()
self.close()
