add --get-all-tags and --get-all-mentions to zfind command

Author: George K. Thiruvathukal

zettelgeist/zdb.py

@@ -129,6 +129,24 @@ def fts_search(self, term_list):
     def fts_query(self, prepared_sql):
         return self.cursor.execute(prepared_sql)
 
+    def get_tags_generator(self):
+        Q = "select distinct(tag) from tags"
+        for row in self.cursor.execute(Q):
+           yield(row['tag'])
+
+    def get_tags_list(self):
+        gen = self.get_tags_generator()
+        return list(gen)
+
+    def get_mentions_generator(self):
+        Q = "select distinct(mention) from mentions"
+        for row in self.cursor.execute(Q):
+            yield(row['mention'])
+
+    def get_mentions_list(self):
+        gen = self.get_mentions_generator()
+        return list(gen)
+
     def done(self):
         self.conn.commit()
         self.conn.close()

zettelgeist/zfind.py

@@ -35,6 +35,12 @@ def get_argparse():
     parser.add_argument(
         '--stats', help="write statistics and parameters to filename", default=None)
 
+    parser.add_argument(
+        '--get-all-tags', help="show all tags in this database (disables all other options)", action="store_true", default=False)
+
+    parser.add_argument(
+        '--get-all-mentions', help="show all mentions in this database (disables all other options)", action="store_true", default=False)
+
     return parser
 
 
@@ -43,6 +49,15 @@ def main():
     args = parser.parse_args()
     argsd = vars(args)
 
+    db = zdb.get(args.database)
+    if args.get_all_tags:
+        print('\n'.join(db.get_tags_list()))
+        exit(0)
+
+    if args.get_all_mentions:
+        print('\n'.join(db.get_mentions_list()))
+        exit(0)
+
     if args.query_prompt:
         input_line = input("zfind> ")
     elif args.query_file:
@@ -56,7 +71,6 @@ def main():
 
     if input_line:
         ast = zquery.compile(input_line)
-        db = zdb.get(args.database)
         gen = db.fts_query(ast[0])
     else:
         print("Warning: Query could not be loaded via --query, --query-file, or --query-prompt")