Browse Source

add auto update function with foolproofing

jason 3 years ago
parent
commit
79385b2055
1 changed files with 7 additions and 2 deletions
  1. 7 2
      api/main.py

+ 7 - 2
api/main.py

@@ -48,15 +48,20 @@ async def formentry(bgtask: BackgroundTasks, name: str = Form(...), phone: str =
 def getdata(db):
     statement = 'SELECT * FROM booking_info ORDER BY time_stamp DESC'
     df = pd.DataFrame()
+    rowcount = 0
 
     for row in db.query(statement):
         date_format = "%Y-%M-%d %H:%M:%S"
-   
         fdate = row['time_stamp'].strftime('%Y-%m-%d')
         
         df = df.append({'填單時間':row['time_stamp'],'姓名':row['name'],'電話':str('\'' + row['phone']),'預約體驗店家':row['location'],'選擇課程':row['course_name'],'預約日期':row['book_date'],'預約時間':row['book_time']}, ignore_index=True)
 
-    save_sheet(df,'紫晴預約資料','預約資料')
+        rowcount +=1
+
+    if rowcount!=0:
+        save_sheet(df,'紫晴預約資料','預約資料')
+    else:
+        print('沒有資料')
 
     return 0