You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.4 KiB
44 lines
1.4 KiB
1 year ago
|
import pymysql
|
||
|
|
from pymysql.cursors import DictCursor
|
||
|
|
import datetime
|
||
|
|
|
||
|
|
db_config = {
|
||
|
|
'host': '192.168.66.101',
|
||
|
|
'user': 'root',
|
||
|
|
'password': 'Sxzgx1209',
|
||
|
|
'database': 'scrapy'
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class DbAction:
|
||
|
|
def __init__(self, end_date='today'):
|
||
|
|
self.conn = pymysql.Connection(**db_config)
|
||
|
|
self.cursor = self.conn.cursor(cursor=DictCursor)
|
||
|
|
self.date_list = self.get_date_list(end_date)
|
||
|
|
|
||
|
|
def get_db_data(self):
|
||
|
|
query_sql = f"SELECT h.id, h.cate, h.date, h.`name`, h.save_link, h.`code`, h.unzip_pwd FROM scrapyh h WHERE h.date IN ({self.date_list});"
|
||
|
|
self.cursor.execute(query_sql)
|
||
|
|
return self.cursor.fetchall()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_date_list(_end_date):
|
||
|
|
date_list = []
|
||
|
|
if _end_date == 'today':
|
||
|
|
date_list.append(f"'{datetime.date.strftime(datetime.date.today(), '%Y-%m-%d')}'")
|
||
|
|
else:
|
||
|
|
start_date = datetime.datetime.today()
|
||
|
|
end_date = datetime.datetime.strptime(_end_date, '%Y-%m-%d')
|
||
|
|
delta = datetime.timedelta(days=1)
|
||
|
|
while start_date > end_date:
|
||
|
|
date_list.append(f"'{datetime.datetime.strftime(start_date, '%Y-%m-%d')}'")
|
||
|
|
start_date = start_date - delta
|
||
|
|
print(date_list)
|
||
|
|
return ','.join(date_list)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
db_action = DbAction('2023-07-28')
|
||
|
|
result = db_action.get_db_data()
|
||
|
|
print(result)
|