首页 > 数据库技术 > 详细

用argparse库去连接mysql

时间:2021-09-02 05:01:10      阅读:11      评论:0      收藏:0      [点我收藏+]
import argparse
import pymysql


class MysqlConnect(object):
    # 连接数据库
    def __init__(self, opt):
        try:
            self.conn = pymysql.connect(
                host=opt.host,
                user=opt.username,
                password=opt.password,
                db=opt.db,
                port=opt.port,
                charset=opt.charset,
            )
        except pymysql.Error as e:
            print(e)

    # 关闭数据库的连接
    def close_conn(self):
        try:
            if self.conn:
                # 关闭连接
                self.conn.close()
        except pymysql.Error as e:
            print(e)

    # 自定义你的数据库操作方法
    def get_data(self):
        # 准备sql
        sql = "select * from machine"
        # 找到cursor
        cursor = self.conn.cursor()
        # 执行sql
        cursor.execute(sql)
        # fetchall全部查
        data = cursor.fetchall()
        # 关闭cursor/连接
        cursor.close()
        self.close_conn()
        return data


if __name__ == ‘__main__‘:
    parser = argparse.ArgumentParser()

    parser.add_argument(‘--host‘, type=str,
                        default=‘***********‘)

    parser.add_argument(‘--port‘, type=int, default=*****)

    parser.add_argument(‘--username‘, type=str, default=‘******‘)

    parser.add_argument(‘--password‘, type=str, default=‘********‘)

    parser.add_argument(‘--db‘, type=str, default=‘*******‘)

    parser.add_argument(‘--charset‘, type=str, default=‘utf8‘)
    opt = parser.parse_args()
    mysql_connect = MysqlConnect(opt)
    data = mysql_connect.get_data()
    print(data)

用argparse库去连接mysql

原文:https://www.cnblogs.com/zhuweidong/p/15203223.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!