def add_to_db(jid, jdate, jvalue, jper):
add_jj = False
add_sum = False
jj = db_session.query(TblJijin).filter(TblJijin.jid == jid,
TblJijin.jdate == jdate).first()
if jj is None:
add_jj = True
jj = TblJijin()
jj.jid = jid
jj.jdate = jdate
jj.jvalue = jvalue
jsum = db_session.query(TblSum).filter(TblSum.jid == jid,
TblSum.jdate == jdate).first()
if jsum is None:
add_sum = True
jsum = TblSum()
jsum.jid = jid
jsum.jdate = jdate
jsum.jper = jper
try:
jsum.jinc = 1 if float(jper) > 0 else 0 if int(jper) == 0 else -1
except Exception as e:
jsum.jinc = 0
try:
if add_jj:
db_session.add(jj)
if add_sum:
db_session.add(jsum)
db_session.commit()
except Exception as e:
db_session.rollback()
db_session.close()
def init_data():
print('init_data...')
from database.db_config import db_session
from database.tbl_admin import TblAdmin
user = TblAdmin()
user.name = "__MAIL__"
user.value = "*****@*****.**"
user.type = 1
email = TblAdmin()
email.name = "__MAILPASSWORD__"
email.value = "xxxxxxxxxx"
email.type = 1
user_exist = db_session.query(
TblAdmin.name).filter(TblAdmin.name == user.name).first()
if user_exist is None:
db_session.add(user)
mail_exist = db_session.query(
TblAdmin.name).filter(TblAdmin.name == email.name).first()
if mail_exist is None:
db_session.add(email)
db_session.commit()
db_session.close()
print("done!")
def drop_less_data(jid):
db_session.query(TblSum).filter(TblSum.jid == jid).delete()
db_session.query(TblJijin).filter(TblJijin.jid == jid).delete()
try:
db_session.commit()
except Exception as e:
print(e)
db_session.rollback()
def init_admin():
from database.tbl_admin import TblAdmin
user = TblAdmin()
user.name = "appversion"
user.value = "1.0"
user.type = 1
db_session.add(user)
db_session.commit()
db_session.close()
def init_jijin():
from database.tbl_jijin import TblJijin
from datetime import datetime
dis = TblJijin()
dis.jid = "test2"
dis.jdate = '2020-05-05'
dis.jvalue = '2.13'
db_session.add(dis)
db_session.commit()
def init_admin():
from database.tbl_admin import TblAdmin
user = TblAdmin()
user.name = "__MAIL__"
user.value = "*****@*****.**"
user.type = 1
db_session.add(user)
db_session.commit()
db_session.close()
def drop_range_date_data():
db_session.query(TblSum).filter(TblSum.jdate >= "2021-10-01",
TblSum.jdate <= "2021-10-07").delete()
db_session.query(TblJijin).filter(TblJijin.jdate >= "2021-10-01",
TblJijin.jdate <= "2021-10-07").delete()
try:
db_session.commit()
except Exception as e:
print(e)
db_session.rollback()
def init_version():
from database.tbl_admin import TblAdmin
version = TblAdmin()
version.name = "appversion"
version.value = "1.0"
version.type = 1
db_session.add(version)
db_session.commit()
db_session.close()
def init_user():
from database.tbl_account import TblAccount
account = TblAccount()
account.username = "******"
account.password = MD5("111111")
account.email = "*****@*****.**"
account.userstate = 0
account.userrole = 0
db_session.add(account)
db_session.commit()
db_session.close()
def clear_history(days):
clear_date = DatetimeManage.get_days_ago(days)
try:
db_session.query(TblBrowsingHistory).filter(
TblBrowsingHistory.browsing_date <= clear_date).delete()
db_session.commit()
db_session.close()
weblog.info("db data history clear...")
except Exception as e:
print(e)
weblog.exception("data history error. {}".format(e))
def init_user():
from database.tbl_account import TblAccount
account = TblAccount()
account.loginname = "youth303"
account.nickname = u"青春"
account.password = MD5("303303")
account.email = ""
account.userstate = 0
account.userrole = 2
db_session.add(account)
db_session.commit()
db_session.close()
def clear_old_data():
pass
data_clear_date = DatetimeManage.get_days_ago(365)
try:
pass
db_session.query(TblJijin).filter(
TblJijin.jdate <= data_clear_date).delete()
db_session.query(TblSum).filter(
TblSum.jdate <= data_clear_date).delete()
db_session.commit()
db_session.close()
except Exception as e:
print(e)
weblog.exception("data TblJijin/TblSum error. {}".format(e))
def init_account():
print('init one account...')
from database.db_config import db_session
from database.tbl_account import TblAccount
from method.data_encode import MD5
account = TblAccount()
account.username = "******"
account.password = MD5("123456")
account.userrole = 0
account.email = "*****@*****.**"
db_session.add(account)
db_session.commit()
db_session.close()
print("add ok")
def init_account():
print('init one account...')
from database.db_config import db_session
from database.tbl_account import TblAccount
from method.data_encode import MD5
account = TblAccount()
account.loginname = "Tornado"
account.nickname = u"飞影"
account.password = MD5("dgj_039103")
account.userrole = 0
account.userstate = 0
account.email = "*****@*****.**"
db_session.add(account)
db_session.commit()
db_session.close()
print("add ok")
def init_setting():
from database.tbl_admin import TblAdmin
db_session.add_all([
TblAdmin(name="currentname", value="feiying", type=0),
TblAdmin(name="description", value="test only", type=0),
TblAdmin(name="admin_email", value="*****@*****.**", type=0),
TblAdmin(name="can_register", value="1", type=1),
TblAdmin(name="can_comment", value="1", type=1),
TblAdmin(name="comments_notify", value="1", type=1),
TblAdmin(name="default_category", value="default_category", type=0),
TblAdmin(name="page_size", value="10", type=1),
TblAdmin(name="rss_size", value="10", type=1),
TblAdmin(name="rss_excerpt", value="1", type=1),
TblAdmin(name="new_rss_size", value="5", type=1),
TblAdmin(name="new_page_size", value="5", type=1),
])
db_session.commit()
def add_data(jid, jdate, jvalue):
tjj = db_session.query(TblJijin).filter(TblJijin.jid == jid,
TblJijin.jdate == jdate).first()
if tjj is None:
tjj = TblJijin()
tjj.jid = jid
tjj.jdate = jdate
tjj.jvalue = jvalue
db_session.add(tjj)
weblog.info("{} {} {} add db.".format(jid, jdate, jvalue))
else:
tjj.jid = jid
tjj.jdate = jdate
tjj.jvalue = jvalue
weblog.info("{} {} {} exist then update.".format(jid, jdate, jvalue))
try:
db_session.commit()
except Exception as e:
weblog.error("{} {} {} add fail. {}".format(jid, jdate, jvalue, e))
def init_info():
from database.db_config import db_session
from database.tbl_admin import TblAdmin
user = TblAdmin()
user.name = "appinfo"
user.value = "1.0"
user.type = 1
user_exist = db_session.query(TblAdmin.name).filter(TblAdmin.name == user.name).first()
if user_exist is None:
db_session.add(user)
user = TblAdmin()
user.name = "appversion"
user.value = "1.0"
user.type = 1
mail_exist = db_session.query(TblAdmin.name).filter(TblAdmin.name == user.name).first()
if mail_exist is None:
db_session.add(user)
db_session.commit()
db_session.close()
def add_sum(jid, jdate, jvalue):
tsum = db_session.query(TblJijin).filter(
TblJijin.jid == jid,
TblJijin.jdate <= jdate).order_by(TblJijin.jdate.desc()).limit(2)
count = tsum.count()
if count != 2:
return None
inc = 0
try:
per = (float(tsum[0].jvalue) - float(tsum[1].jvalue)) / float(
tsum[1].jvalue)
if per > 0: inc = 1
if per < 0: inc = -1
per = round(per * 100, 3)
per = str(per)
except Exception as e:
per = "--"
tas = db_session.query(TblSum).filter(TblSum.jid == jid,
TblSum.jdate == jdate).first()
if tas is None:
tas = TblSum()
tas.jdate = jdate
tas.jid = jid
tas.jper = per
tas.jinc = inc
db_session.add(tas)
cjjlog.info("sum: {} {} {} add db.".format(jid, jdate, per))
else:
tas.jdate = jdate
tas.jid = jid
tas.jper = per
tas.jinc = inc
cjjlog.info("sum: {} {} {} exist then update.".format(jid, jdate, per))
# print(cjjlog)
# print("sum: {} {} {} exist then update.".format(jid, jdate, per))
try:
db_session.commit()
except Exception as e:
cjjlog.error("sum: {} {} {} add fail. {}".format(jid, jdate, per, e))
TblAdmin(name="description", value="test only", type=0),
TblAdmin(name="admin_email", value="*****@*****.**", type=0),
TblAdmin(name="can_register", value="1", type=1),
TblAdmin(name="can_comment", value="1", type=1),
TblAdmin(name="comments_notify", value="1", type=1),
TblAdmin(name="default_category", value="default_category", type=0),
TblAdmin(name="page_size", value="10", type=1),
TblAdmin(name="rss_size", value="10", type=1),
TblAdmin(name="rss_excerpt", value="1", type=1),
TblAdmin(name="new_rss_size", value="5", type=1),
TblAdmin(name="new_page_size", value="5", type=1),
])
db_session.commit()
if __name__ == "__main__":
# init_admin()
# init_user()
# init_setting()
from database.tbl_discuss import TblDiscuss
from datetime import datetime
dis = TblDiscuss()
dis.discuss = u"sub discuss"
dis.author = 3
dis.status = 0
dis.original = 1
dis.topic_id = 1
db_session.add(dis)
db_session.commit()
def save(self):
db_session.add(self)
db_session.commit()
