1 import pymysql
2 import time
3
4 def logger(msg):
5 print("[%s] %s" % (time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), msg))
6
7 host = 'xxxxx'
8 user = 'xxxxxxx'
9 password = 'xxxx'
10 database = 'xxxx'
11 # 打开数据库连接
12 mysql_connect = pymysql.connect(host=host, user=user, password=password, db=database, port=3306, charset="utf8")
13
14 if __name__ == '__main__':
15 sql = "SELECT * FROM marbleuser.article where content_type=1;"
16 # 使用cursor()方法获取操作游标
17 cursor = mysql_connect.cursor()
18 logger('开始查询')
19 # 使用execute方法执行SQL语句
20 cursor.execute(sql)
21 # 使用 fetchone() 方法获取一条数据
22 # 使用 fetchall() 方法获取所有数据
23 values = cursor.fetchall()
24 mysql_connect.commit()
25 logger('开始写到txt')
26 for i in values:
27 article_id = i[0]
28 with open(file='articleId.txt', mode='a+', encoding='utf-8') as f:
29 f.writelines(article_id + '\n')
30 #关闭数据库连接
31 mysql_connect.close()
32 logger('运行结束')