python数据库封装调用

在写脚本的过程，考虑多个脚本要用到数据库，如果每次都去写数据库的基本信息，就会显得很冗余，另外这也不利于维护。

basesql.py

import pymysql

class DBInfo(object):
    # 数据库基本信息
    CONN_base = pymysql.connect(
        host='192.XXX.XXX.XX',
        port=XXXX, #端口号
        user='username',
        password='xxxx',
        database='xxxxx',#库名
        charset='utf8'  
    )
    CONN_base0 = pymysql.connect(
        host='192.XXX.XXX.XX',
        port=XXXX, #端口号
        user='username',
        password='xxxx',
        database='xxxxx',#库名
        charset='utf8'  
    )

main.py

from basesql import DBInfo
import pymysql
class P_Sql(object):
    def __init__(self,type="1"):
        self.conn = DBInfo.CONN_base
        if type=="2":
            self.conn = DBInfo.CONN_base0

    def run(self,code):
        cur=self.conn.cursor()
        #查询语句
        sql = "SELECT factory,iphone from xxx  where code ='%s'" %(code)
        #调用游标的execute()方法执行sql语句
        cur.execute(sql)
        #使用fetchall()获取全部数据，即获取查询结果
        result = cur.fetchall()
        if (len(result) == 0):
            print("该用户不存在")
        else:
            #依次输出查询的记录
            if(ord(result[0][0])==0):
                print(code+"用户无工厂,手机号码是"+result[0][1])
            else:
                print(code+"用户有工厂,手机号码是"+result[0][1])

if __name__ == "__main__":
     P_Sql(type="2").run(code="1392010")