fix issue when long_url not save

hitnik · Aug 21, 2021 · 229e536 · 229e536
1 parent e5727ca
commit 229e536
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/app/utils.py b/app/utils.py
@@ -98,6 +98,9 @@ def save_url(short, long):
             if long_url_exist(long):
                 long_inst = get_long_url_from_db(url=long)
                 insert_short_url(short, long_inst[0])
+            else: 
+                long_id = insert_long_url(long)
+                insert_short_url(short, long_id)
             return short
         else:
             raise URLExistsError