o

ig_tags/ig_selenium.py

@@ -14,9 +14,11 @@ from selenium.webdriver.support import expected_conditions as ec
 # import dataset
 import pymysql
 pymysql.install_as_MySQLdb()
+
 import pd
 import setting
 
+
 account = ['chenlinrain','enjoylisteningswift','novak_goodenough']
 
 password = ['weareyoung12']
@@ -84,7 +86,7 @@ def download_post():
 
 
 def hashtag():
-    input_hashtag = 'pet_food'
+    input_hashtag = 'pet_cake'
     collect_hashtag = []
     c=0
     result = download_post()[0]
@@ -113,28 +115,28 @@ def hashtag():
             c=0
     driver.close()
     print(collect_hashtag)
-    db = pymysql.connect(host='db.ptt.cx',
-                         user='choozmo',
-                         password='pAssw0rd',
-                         database='seo')
-    cur = db.cursor()
+    db_company = pymysql.connect(host='db.ptt.cx',
+                                 user='choozmo',
+                                 password='pAssw0rd',
+                                 database='seo')
+    cur = db_company.cursor()
     query_new_col = f"ALTER TABLE seo.ig_pet ADD COLUMN {input_hashtag} VARCHAR(45) NULL"
     cur.execute(query_new_col)
-    id_number = 1
-    cur.execute("select * from ig_tags.new_table order by id desc limit 1")
+    id_number = 0
+    cur.execute("select * from seo.ig_pet order by `index` desc limit 1")
     last_id = cur.fetchall()[0][0]
     insert_row = len(collect_hashtag)-last_id
     for i in range(0, insert_row):
-        query_insert = f"INSERT INTO ig_tags.new_table ({input_hashtag}) VALUES ('')"
+        query_insert = f"INSERT INTO seo.ig_pet ({input_hashtag}) VALUES ('')"
         cur.execute(query_insert)
     for i in collect_hashtag:
         i = i.replace('#', '')
-        query_update = f"UPDATE ig_tags.new_table SET {input_hashtag}='{i}' where id='{id_number}'"
+        query_update = f"UPDATE seo.ig_pet SET {input_hashtag}='{i}' where `index`='{id_number}'"
         cur.execute(query_update)
         id_number += 1
 
-    db.commit()
-    db.close()
+    db_company.commit()
+    db_company.close()
     return hashtag