DRIVER = 'pymysql'
USERNAME = "admin"
PASSWORD = "admin+123"  # 每個人設置的名字和賬號會不同，這里是自己設定的賬號密碼
HOST = '127.0.0.1'
PORT = '3306'
DATABASE = 'testdb'  # 這里是數(shù)據(jù)庫文件名
SQLALCHEMY_DATABASE_URI = '{}+{}://{}:{}@{}:{}/{}?charset=utf8'.format(DIALECT, DRIVER, USERNAME, PASSWORD, HOST, PORT,
                                                                       DATABASE)
SQLALCHEMY_TRACK_MODIFICATIONS = True
SQLALCHEMY_ECHO = True

from flask_sqlalchemy import SQLAlchemy
db = SQLAlchemy()

from flask import Flask
from config.mysqlconfig import db
from models import usermodel,assetmodel  # 在調(diào)用create_all()之前，必須導入需要操作的模型，否則不會創(chuàng)建表
app = Flask(__name__) # 初始化flask應用
app.config.from_object('config.config')
db.init_app(app) #使用app對sqlalchemy進行初始化綁定
with app.app_context():
    db.create_all() # 在調(diào)用create_all()之前，必須導入需要操作的模型，否則不會創(chuàng)建表
from api.userapi import user_blueprint
from api.testdataapi import testdata_blueprint
app.register_blueprint(user_blueprint)  # 將管理員的藍圖注冊到應用中
app.register_blueprint(testdata_blueprint)  # 將測試數(shù)據(jù)管理的藍圖注冊到應用中

from config.mysqlconfig import db
from views.responsehandler import *
class AssetModel(db.Model):
    __tablename__ = 'asset'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    init_date = db.Column(db.DateTime, nullable=False)
    user_name = db.Column(db.String(20), unique=True, nullable=False)
    fund_account = db.Column(db.String(20), nullable=False)
    fund_days = db.Column(db.String(30), nullable=False)
    fund_money = db.Column(db.String(20), nullable=False)
    update_time = db.Column(db.DateTime, nullable=False)
    def __init__(self, user_name, init_date,fund_account, fund_days, fund_money, update_time):
        self.user_name = user_name
        self.init_date = init_date
        self.fund_account = fund_account
        self.fund_days = fund_days
        self.fund_money = fund_money
        self.update_time = str(update_time)

@testdata_blueprint.route('/add_data', methods=['POST'])
def add_data():
    username = request.form.get('username')
    fund_account = request.form.get("account")
    days = request.form.get("days")
    asset = request.form.get("asset")
    try:
        fund_account = int(fund_account)
        days = int(days)
        asset = float(asset)
        total_asset = days * asset
        init_date = update_time=time.strftime("%Y-%m-%d",time.localtime())
    except Exception as e:
        print(e)
    asset_data = AssetModel(init_date=init_date,user_name=username,fund_account=fund_account,fund_days=days,fund_money=total_asset,update_time=time.strftime("%Y-%m-%d %H:%M:%S",time.localtime()) )
    db.session.add(asset_data)
    db.session.commit()