数据库连接

在pycharm中安装pymysql

# 导入pymysql模块
import pymysql as ps
def conMysql():
    try:
        # 创建连接
        con = ps.connect(user = "root", password = "1234", host = "localhost", database = "company", port = 3306, charset = 'utf8')

    except ps.err.OperationalError:
        print("连接失败，请检查连接参数")
    else:
        print("创建游标")
        # 创建游标
        cur = con.cursor()
        # cur.callproc('存储过程名称',('参数'))     # 调用存储过程
        return con, cur

def closeCon(con, cur):
    con.close()
    cur.close()

if __name__ == '__main__':
    sql = 'select * from userinfo'
    con, cur = conMysql()           # 调用连接函数
    count = cur.execute(sql)        # 执行sql
    # data_content1 = cur.fetchone()  # 打印一条数据
    data_content2 = cur.fetchall()  # 打印全部数据,存储在一个元祖中
    print(count,data_content2)
    closeCon(con, cur)              # 关闭游标和连接