BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(sys.argv[0])))
sys.path.append(BASE_DIR)
from sqlite3 import dbapi2 as sqlite
from sqlalchemy import create_engine, Column, String, Integer, BigInteger
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
# 这些都不用管
Base = declarative_base()
engine = create_engine('sqlite+pysqlite:///%s'%os.path.join(data_path, 'standard.db'), module=sqlite)
session_obj = sessionmaker(bind=engine)
engine2 = create_engine("mysql+%s://%s:%s@%s:%d/%s?charset=%s" % ('pymysql', 'root', '111111', '127.0.0.1', 3306, 'standard', 'utf8'), max_overflow=5)
session_obj2 = sessionmaker(bind=engine2)
# 上面连接了2个数据库,一个是mysql,一个是sqlite3,值得注意的是,连接mysql使用的是pymysql模块,也就是说,sqlalchemy底层拼接字符串调用的是pymysql的方法
# 另一点是,不管是什么数据库,只要这个ORM支持,用的语法都一样!不管是mysql还是sqlite甚至Orcale(但就是没有微软全家桶)
# 封装一个表
class Enterprise(Base):
__tablename__ = 'enterprise'
id = Column(Integer, primary_key=True)
enterpriseName = Column(String(250), unique=True) # 企业名称
orgCode = Column(String(18), unique=True) # 信用代码
areaCode = Column(Integer, index=True) # 行政区划代码
xzqh = Column(String(50), index=True) # 行政区划
fddbr = Column(String(50)) # 法人
jgdz = Column(String(250)) # 地址
...
#这个class就是一个表,这是ORM的核心,把表转化成类,表里的字段使用类的属性来对应
# 向表里插入数据示例:
"""
例如我有一个数据是字典形式
dct = {
'enterpriseName': '123',
...
}
"""
session = session_obj() # 同样有游标,这点上我觉的Django的ORM更友好
obj1 = Enterprise(**dct)
session.add(obj1)
session.commit() # 支持事务
# 一个sql语句也没写,只要会python语法,就能用ORM