import pymysql
import redis
import configparser
def redis_conn() -> redis.Redis:
ip, port, redis_db = get_conn_ip('redis', 'redis_config', 'redis_ip', 'redis_port')
pool = redis.ConnectionPool(host=ip, port=port, db=redis_db)
return redis.Redis(connection_pool=pool)
def mysql_conn():
info = get_conn_ip('mysql', 'mysql_config', 'mysql_ip', 'mysql_port')
return pymysql.connect(host=info[0], port=info[1], user=info[2], password=info[3], database=info[4], charset='utf8mb4')
def mysql_read(read_sql):
conn = mysql_conn()
cursor = conn.cursor()
cursor.execute(read_sql)
result = cursor.fetchall()
cursor.close()
conn.close()
return result
def mysql_write(write_sql):
conn = mysql_conn()
cursor = conn.cursor()
cursor.execute(write_sql)
conn.commit()
cursor.close()
conn.close()
def get_conn_ip(data_type, config, ip, port, user=None, password=None, database=None):
cfg = configparser.ConfigParser()
cfg.read('conn_ip.ini')
if data_type == "redis":
ip = cfg[config][ip]
port = cfg[config][port]
redis_db = cfg[config][database]
return ip, port, redis_db
elif data_type == "mysql":
ip = cfg[config][ip]
port = cfg[config][port]
user = cfg[config][user]
password = cfg[config][password]
database = cfg[config][database]
return [ip, port, user, password, database]
if __name__ == '__main__':
redis_conn()
[redis_config]
redis_ip = '127.0.0.1'
redis_port = 1101
redis_db = 0
[mysql_config]
mysql_ip = '127.0.0.1'
mysql_port = 3300
user = 'laowang'
password = '123'
mysql_db = 'test'