Python连接数据库(登录)

import pymysql

def mysql(username, password):
    #这里我设帐号和密码长度有所限制

    if len(str(username)) != 16 or len(str(password)) != 6:
        print("帐号或密码错误")
        return
    else:
        #连接本地数据库

        conn = pymysql.connect(host='127.0.0.1',
                               user='root',
                               passwd='你的数据库密码',
                               db='mysql',
                               charset='utf8')
        cur = conn.cursor()
        #执行数据库语句(使用atm这个数据库)

        cur.execute("USE atm")
        #执行数据库语句(查询login表中符合要求的所有信息)

        login = cur.execute("select * from login where username=%s and password=%s", (username, password))
        #提交数据库请求

        conn.commit()

        try:
            if login:
                print("OK")
        except:
            print("no OK")
        finally:
            cur.close()
            conn.close()

mysql(1234567891234567, 123456)