partsdb

commit eee65dd7c1e7094e5b0f8f08b32538d162aa18b4
parent ce839b37cd28c30a98d014aa1dba1f0f149b66f3
Author: Paco Esteban <paco@e1e0.net>
Date:   Sat, 13 Mar 2021 10:43:39 +0100

implement list categories

Diffstat:
M
myChips.py
 | 
11
+++++++++++

1 file changed, 11 insertions(+), 0 deletions(-)
diff --git a/myChips.py b/myChips.py
@@ -125,6 +125,13 @@ def adjust_stock(part_id, stock_mod):
 def export_db(dest_folder):
     pass
 
+def list_categories():
+    categories = db.get_categories()
+    print("ID\tName")
+    print("-"*79)
+    for c in categories:
+        print(f"{c[0]}\t{c[1]}")
+
 
 if __name__ == '__main__':
     ap = argparse.ArgumentParser()
@@ -135,6 +142,8 @@ if __name__ == '__main__':
     # add
     ap_add = asp.add_parser("add", help="Add new part from Octopart")
     ap_add.add_argument("mpn", help="Manufacturer part number")
+    # cat
+    ap_cat = asp.add_parser("cat", help="List categories")
     # list
     ap_list = asp.add_parser("list", help="List all parts from a category")
     ap_list.add_argument("category", help="Category Name")
@@ -175,6 +184,8 @@ if __name__ == '__main__':
         adjust_stock(args.part_id, args.stock_mod)
     elif args.command == 'export':
         export_db(args.dest_folder)
+    elif args.command == 'cat':
+        list_categories()
     else:
         ap.print_help()
         sys.exit(0)