Selaa lähdekoodia

更新get_spread

zooeytsai 2 vuotta sitten
vanhempi
commit
8237921434
1 muutettua tiedostoa jossa 8 lisäystä ja 4 poistoa
  1. 8 4
      INNNews/get_spread.py

+ 8 - 4
INNNews/get_spread.py

@@ -14,6 +14,7 @@ import datetime
 from gspread_pandas import Spread, Client
 from oauth2client.service_account import ServiceAccountCredentials
 import os
+from itertools import islice
 
 def save_sheet(df,filename,tabname,startpos='A1'):
 
@@ -38,20 +39,23 @@ def get_sheets(filename):
     spread = Spread(filename,creds=credentials)
 
     df=spread.sheet_to_df(sheet='2022')
-    for idx,row in df.iterrows():
+    for idx, row in islice(df.iterrows(), 1, None):
         r1=row[1].strip()
         if len(r1)>1:
             r2=row[2].strip()
             r3=row[3].strip()
-            result.append({'kw':r1+" "+r2+" "+r3})
+            if "病假" not in r1:
+                result.append({'kw':r1+" "+r2+" "+r3})
+            else:
+                pass
     return result
 
 result=get_sheets('引新聞INNEWS稿單xSEO優化')
 js=json.dumps(result, ensure_ascii=False)
 
 print(js)
-r = redis.Redis(host='db.ptt.cx', port=6379, db=1,password='choozmo9')
-r.set('innews_sheet',js)
+# r = redis.Redis(host='db.ptt.cx', port=6379, db=1,password='choozmo9')
+# r.set('innews_sheet',js)
 
 #    spread.df_to_sheet(df, index=False, sheet=tabname, start=startpos, replace=False)
 #