partsdb

commit a65ce238b17cfd63df520699fa75d691edbddaaa
parent cffded8bbd2890aa31caf2b13bbafd59586f51b3
Author: Paco Esteban <paco@e1e0.net>
Date:   Sat, 30 Oct 2021 19:28:18 +0200

rename format to output on list and add it to search too

Diffstat:
M
partsdb/helpers.py
 | 
8
++++----
M
partsdb/partsdb.py
 | 
13
++++++++-----

2 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/partsdb/helpers.py b/partsdb/helpers.py
@@ -8,12 +8,12 @@ import tempfile
 import time
 
 
-def print_parts_list(parts, format='full'):
-    if format == 'short':
+def print_parts_list(parts, output='full'):
+    if output == 'short':
         _list_ascii_short(parts)
-    if format == 'full':
+    if output == 'full':
         _list_ascii(parts)
-    if format == 'json':
+    if output == 'json':
         _list_json(parts)
 
 
diff --git a/partsdb/partsdb.py b/partsdb/partsdb.py
@@ -134,14 +134,14 @@ def list_parts(category, short):
     helpers.print_parts_list(parts, short)
 
 
-def search_part(search_term):
+def search_part(search_term, output):
     parts = db.search_parts(search_term)
 
     if not parts:
         print("No parts found")
         return
 
-    helpers.print_parts_list(parts)
+    helpers.print_parts_list(parts, output)
 
 
 def get_part(part_id):
@@ -229,12 +229,15 @@ def main():
     ap_list = asp.add_parser("list",
                              help="List all parts from a category (or all)")
     ap_list.add_argument("category", help="Category Name")
-    ap_list.add_argument("-f", dest='format',
+    ap_list.add_argument("-o", dest='output',
                          choices=['full', 'short', 'json'], default='full',
                          help="Short output")
     # search
     ap_search = asp.add_parser("search", help="Search for parts")
     ap_search.add_argument("search_term", help="Term to search for")
+    ap_search.add_argument("-o", dest='output',
+                           choices=['full', 'short', 'json'], default='full',
+                           help="Short output")
     # get
     ap_get = asp.add_parser("get", help="Get all details for a part")
     ap_get.add_argument("part_id", help="Part Id", type=int)
@@ -264,9 +267,9 @@ def main():
         add_part(args.mpn, args.quantity, args.category, args.storage,
                  args.type)
     elif args.command == 'list':
-        list_parts(args.category, args.format)
+        list_parts(args.category, args.output)
     elif args.command == 'search':
-        search_part(args.search_term)
+        search_part(args.search_term, args.output)
     elif args.command == 'get':
         get_part(args.part_id)
         if args.datasheet: