Forráskód Böngészése

email duplication check

ming 3 éve
szülő
commit
75bc2731d8
2 módosított fájl, 9 hozzáadás és 4 törlés
  1. BIN
      api/__pycache__/main.cpython-39.pyc
  2. 9 4
      api/main.py

BIN
api/__pycache__/main.cpython-39.pyc


+ 9 - 4
api/main.py

@@ -13,6 +13,7 @@ import pandas as pd
 from gspread_pandas import Spread, Client
 from oauth2client.service_account import ServiceAccountCredentials
 import gspread
+from first import first
 
 app = FastAPI()
 
@@ -50,12 +51,16 @@ async def showdata(limit):
 
 @app.post("/add_sub")
 async def add_subclient_info(req:subreq):
-    code = 'ok'
+    code = 1
     sub_table = db['Subscribe']
     time_stamp = datetime.fromtimestamp(time.time()).strftime("%Y-%m-%d %H:%M:%S")
-    pk = sub_table.insert({'email':req.email,'timestamp':time_stamp})
-    make_ci_df()
-    return {'msg':code}
+
+    statement='SELECT COUNT(1) FROM Subscribe WHERE email = "'+req.email+'"'
+    if first(db.query(statement))['COUNT(1)']==0:
+        pk = sub_table.insert({'email':req.email,'timestamp':time_stamp})
+        make_ci_df()
+        code = 0
+    return {'msg':code}#0 means succss added, 1 means email is duplicated
 
 @app.get("/get_sub_list")
 async def add_subclient_info():