Merge pull request #8 from link-shrink/implement-milestones

milestones

firestoreDB/__init__.py

@@ -2,3 +2,4 @@
 from .save_firestore import save_firestore
 from .get_firestore import get_firestore
 from .get_firestore_where import get_firestore_where
+from .increase_firestore import increase_firestore

firestoreDB/increase_firestore.py

@@ -0,0 +1,7 @@
+from firebase_admin import firestore
+from .init_firestore import db
+
+
+async def increase_firestore(field: str, quantity: int):
+    doc_ref = db.collection("milestones").document("links")
+    doc_ref.update({field: firestore.Increment(quantity)})

link/post_data.py

@@ -1,5 +1,6 @@
 import firestoreDB
 import validator
+import milestones
 
 
 async def post_data(original_link: str, link_id: str):
@@ -8,5 +9,9 @@ async def post_data(original_link: str, link_id: str):
 
     data = {"link_id": link_id, "original_link": original_link}
 
+    await milestones.increase_quantity()
+    await milestones.increase_short_length(f"keskn.uz/{link_id}")
+    await milestones.increase_original_length(original_link)
     await firestoreDB.save_firestore("links", link_id, data)
+
     return data

main.py

@@ -20,7 +20,7 @@ def get_response(data):
     return {
         "ok": True,
         "link_id": link_id,
-        "short_link": f"keskn.uz/{link_id}",
+        "short_link": f"https://keskn.uz/{link_id}",
         "original_link": data["original_link"],
     }
 
@@ -31,7 +31,7 @@ async def get_link(link_id):
     if len(data) == 0:
         return {"ok": False, "message": "Link not found"}
 
-    return data[0]
+    return get_response(data[0])
 
 
 class CreateLink(BaseModel):

milestones/__init__.py

@@ -0,0 +1,3 @@
+from .quantity import increase_quantity
+from .short_length import increase_short_length
+from .original_length import increase_original_length

milestones/original_length.py

@@ -0,0 +1,5 @@
+import firestoreDB
+
+
+async def increase_original_length(original_link: str) -> None:
+    await firestoreDB.increase_firestore("original_length", len(original_link))

milestones/quantity.py

@@ -0,0 +1,5 @@
+import firestoreDB
+
+
+async def increase_quantity() -> None:
+    await firestoreDB.increase_firestore("quantity", 1)

milestones/short_length.py

@@ -0,0 +1,5 @@
+import firestoreDB
+
+
+async def increase_short_length(short_link: str) -> None:
+    await firestoreDB.increase_firestore("short_length", len(short_link))