partsdb

commit 2e2f7748670a5920db3c47b8a597a7ab238368a7
parent e0cd83e8c3545d1c7766d25b8973ce01f64adf72
Author: Paco Esteban <paco@e1e0.net>
Date:   Thu, 19 Oct 2023 18:28:09 +0200

print id and pn when adding a new part

Diffstat:
M
partsdb/database.py
 | 
1
+
M
partsdb/partsdb.py
 | 
10
+++++++++-

2 files changed, 10 insertions(+), 1 deletion(-)
diff --git a/partsdb/database.py b/partsdb/database.py
@@ -204,6 +204,7 @@ class PartsDB:
                 part["part_type"],
             )
             c.execute(INSERT_PART_QUERY_FROM_TMPL, update_tuple)
+            return c.lastrowid
 
     def get_part(self, part_id):
         c = self.conn.cursor()
diff --git a/partsdb/partsdb.py b/partsdb/partsdb.py
@@ -122,6 +122,9 @@ def add_part_from_octo(mpn, quantity, category, storage, part_type):
     new_id = db.new_part(part)
     db.new_part_history_event(new_id, quantity, "first purchase")
 
+    if new_id:
+        print(f"New part created with ID: {new_id} and PN: {new_id:04X}")
+
 
 def add_part_from_template():
     new_part_fields = {
@@ -145,7 +148,12 @@ def add_part_from_template():
         subprocess.call([editor, fp.name])
         try:
             new_part = yaml.safe_load(fp)
-            db.new_part_from_template(new_part)
+            new_id = db.new_part_from_template(new_part)
+            if new_id:
+                db.new_part_history_event(
+                    new_id, new_part["quantity"], "first purchase"
+                )
+                print(f"New part created with ID: {new_id} and PN: {new_id:04X}")
         except yaml.YAMLError as exc:
             print(exc)