def get_common_row(stock_id, col_name, db_session=db_info, db=Info, p=__proc1):
a = FmUtils()
trans = a.load_from_json("./json/trans.json")
trans = a.swap_dict(trans)
name = a.load_from_json("./json/table_trans.json")
name = a.swap_dict(name)
exclude = ['revenue']
if col_name == "date":
decor = [False, "", False]
elif col_name in exclude:
decor = [False, "Y", False]
else:
decor = a.load_from_json("./json/decor.json")
decor = decor[trans[col_name]]
if decor[1] == "%":
decor[1] = "\\%"
print(col_name)
print(decor)
res = db_session.query(getattr(db, col_name)).filter(
db.stock_id == stock_id).order_by(desc(db.date)).all()
if col_name == "date":
line = ""
elif col_name in exclude:
line = name[col_name]
res.pop()
else:
line = trans[col_name]
# check if exists
for i in res:
if col_name == "date":
line = line + "& " + str(i[0]) + decor[1] + " "
else:
line = line + "& " + str(p(i[0])) + decor[1] + " "
line += "\\\\"
if decor[0]:
line = "\\rowcolor[gray]{0.9}" + line
if decor[2] == 'dot':
line = line + "\\hdashline"
if decor[2] == 'solid':
line = line + "\midrule"
return (line)
from fm_utils import FmUtils
from db_setup import Info,db_info,Concise_Table
from sqlalchemy import exists
from pprint import pprint
import os
a = FmUtils()
for ii in range(2012,2018):
stock_name = '大华股份'
mode = 1
# -------------init -------------------
b = a.load_from_json("./"+stock_name+"/info.json")
key = stock_name+str(ii)+'.pdf'
v = b[key]
if v['mode']!= 99:
mode = v['mode']
path = os.path.join('./',stock_name,key)
# dirc = './' + stock_name + '/'
# pprint(dirc+key)
print(path)
d = a.read_and_clean(path,v['table_pages'],mode = mode)
pprint(d)
# ------------ init json ------------
name = a.load_from_json("./json/table_trans.json")
json_loc = dirc + stock_name + str(int(v['year'])-1)+'.json'
if os.path.exists(json_loc):
set_last = a.load_from_json(json_loc)
print("found")
else:
set_last = a.load_from_json("./json/reports_template.json")
def write_tex(path, stock_id, flag_rotate=0):
a = FmUtils()
trans = a.load_from_json("./json/trans.json")
num_rec = db_info.query(Info).filter(Info.stock_id == stock_id).count()
stock_name = db_info.query(
Info.stock_name).filter(Info.stock_id == stock_id).all()[0][0]
row_one = get_common_row(stock_id, "date")
row_two = ""
tab_title = str(stock_id + " " + stock_name + " 基本信息")
for i in range(1, num_rec + 1):
row_two = row_two + "\cmidrule(l){" + str(i + 1) + "-" + str(i +
1) + "}"
with open(path, 'w') as f:
f.write("\\documentclass[a4paper,12pt]{article} \n")
f.write("\\usepackage{booktabs}\n")
f.write("\\usepackage{colortbl}\n")
f.write("\\usepackage{tabu}\n")
f.write("\\usepackage{xeCJK}\n")
f.write("\\usepackage{adjustbox}\n")
f.write("\\usepackage{rotating}\n")
f.write("\\usepackage{fontspec} \n")
f.write("\\setCJKmainfont[Scale=0.8]{SimSun} \n")
f.write("\\setCJKmonofont{SimSun}\n")
f.write(
"\\setmainfont[Scale=0.85, Ligatures={Required,Common,Contextual,TeX}]{TeX Gyre Schola} \n"
)
#---------------
f.write("\n\n\n")
#---------------
f.write("%dashed line\n")
f.write("\\usepackage{array}\n")
f.write("\\usepackage{arydshln}\n")
f.write("\\setlength\\dashlinedash{0.9pt}\n")
f.write("\\setlength\\dashlinegap{1.5pt}\n")
f.write("\\setlength\\arrayrulewidth{0.3pt}\n")
f.write("\\begin{document}\n")
if (flag_rotate):
f.write("\\begin{sidewaystable}[!htbp]\n")
else:
f.write("\\begin{table}[!htbp]\n")
f.write("\\centering\n")
f.write("\\caption{" + tab_title + "}\\label{tab:aStrangeTable}\n")
# if(flag_rotate):
# f.write("\\begin{adjustbox}{angle=90}\n")
seg = "l" + "r" * (num_rec)
f.write("\\begin{tabular}{" + seg + "}\n")
f.write("\\toprule\n")
f.write(row_one + "\n")
f.write(row_two + "\n")
for i in trans:
f.write(get_common_row(stock_id, trans[i]) + "\n")
f.write("\\midrule\n")
f.write(
get_common_row(stock_id, 'revenue', db_info, Concise_Table,
__proc2) + "\n")
f.write("\\bottomrule\n")
f.write("\\end{tabular}\n")
if (flag_rotate):
# f.write("\\end{adjustbox}\n")
f.write("\\end{sidewaystable}\n")
else:
f.write("\\end{table}\n")
f.write("\\end{document}\n")