连接
import psycopg2
import redis
conn = psycopg2.connect(database="***",user="****",password="****", host="192.168.1.3", port="5432")
rds_pool = redis.ConnectionPool(host="192.168.1.16", port=6379, db=10, password="****")
rds_kwai_live = redis.StrictRedis(connection_pool=rds_pool)
## 建立游标,用来执行数据库操作
cursor = conn.cursor()
## 执行SQL命令
sql = """select distinct room_id from kwai.tar_shop where insert_time > '2020-08-31 12:00:00';"""
cursor.execute(sql)
## 提交SQL命令
conn.commit()
## 获取SELECT返回的元组
rows = cursor.fetchall()
# print(rows)
for row in rows:
rds_kwai_live.sadd('kwai_live_shop_room_id', row[0])
print(row)
## 关闭游标
cursor.close()
## 关闭数据库连接
conn.close()