安装pyMySQL>> pip install pyMySQL
数据连接示例#! /usr/bin/env python
# -*- coding: utf-8 -*-
"""
@author: Oscar
@file: gen_random_db
@time: /11/19 14:15
"""
import string
import random
import pymysql
def gen_rand_str():
str_digits = string.digits
str_letters = string.ascii_letters
rand_str = str_digits + str_letters
rand_str = random.sample(rand_str, 10)
return ''.join(rand_str)
def db_conn():
conn = pymysql.connect("192.168.0.5", "admin", "admin", "test")
return conn
def handle_table(conn, table_name):
try:
cursor = conn.cursor() # 获取数据库游标
sql = "select id, random_code from " + table_name + " where random_code is null"
update_sql = "update " + table_name + " set random_code = '%s' where id = '%s' "
cursor.execute(sql) # 执行sql
results = cursor.fetchall() # 获取所有表记录
for row in results: # 遍历数据表中的每一行
id, code = row; # 获取行数据
if code == None:
code = gen_rand_str()
cur = conn.cursor()
cur.execute(update_sql % (code, id))
cur.close()
cursor.close()
mit()
except :
print("error!")
if __name__ == '__main__':
conn = db_conn()
handle_table(conn, "tb_test")
conn.close()