【python案例】将生成的优惠券放入到mysql中

#!/usr/bin/python
# -*- coding: utf-8 -*-
import base64
import re
from sqlalchemy import Column, String, DATE, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
import sys
# 初始化编码
reload(sys)
sys.setdefaultencoding('utf-8')
 
# 创建对象的基类
Base = declarative_base()
 
# 初始化数据库连接参数
database_info = {
    'user': 'root',
    'passwd': '',
    'ip': 'localhost',
    'port': '3306',
    'database': 'spider'
}
 
# 定义Coupon对象
class Coupon(Base):
    # 表的名字
    __tablename__ = 'coupon'
 
    # 表的结构
    id = Column(String(200), primary_key=True)
    deadline = Column(DATE)
    userID = Column(String(200))
    code = Column(String(200))
 
 
def make_connect(DB_info):
    # 初始化数据库连接
    connect_str = 'mysql+pymysql://{user}:{passwd}@{ip}:{port}/{database}'.format(**DB_info)
    engine = create_engine(connect_str)
    # 创建DBSession类型:
    DBSession = sessionmaker(engine)
    #创建Session
    session = DBSession()
    return session
 
#解析优惠码
def parse_coupon(c_code):
    return base64.urlsafe_b64decode(c_code.encode('utf-8'))
 
 
def upload_to_database():
    #连接数据库
    session = make_connect(database_info)
    #打开上一个文件生成优惠码文件
    with open('../coupon/coupon.txt', 'r') as file:
        for line in file.readlines():
            c_id = re.findall(r'.*,.*:(.*)', str(parse_coupon(line)))
            session.add(Coupon(id=c_id.pop(), code=line))
        session.commit()
        session.close()
 
if __name__ == '__main__':
    upload_to_database()