tfse-app-api-v0.2/input/scripts.py

183 lines
5.1 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import copy
import time
from common.scripts import read_json_file, file_path, make_id
from input.db import *
from user.db import find_user_info
from user.scripts import check_verified
def get_current_year():
"""
获取当前年度
Parameters:
-
Returns:
current_year 当前年度
"""
current_year = time.strftime("%Y", time.localtime()) + ""
return current_year
def make_3_report_dates():
"""
生成最近三年报告期 日期
Parameters:
-
Returns:
report_dates 三年报告期
"""
current_year = int(time.strftime("%Y", time.localtime()))
date_01 = str(current_year) + '-12-31'
date_02 = str(current_year-1) + '-12-31'
date_03 = str(current_year-2) + '-12-31'
report_dates = [date_01, date_02, date_03]
return report_dates
def gen_new_rid():
"""
生成新的评价ID如果该ID存在则重新生成
Parameters:
-
Returns:
new_id: 生成的企业ID
"""
# 生成新ID
new_id = make_id()
# 检查新ID是否存在如果存在则继续生成新ID
case = find_rating_records({"评价ID": new_id}) is []
while case:
new_id = make_id()
# 新ID可使用返回新ID
return new_id
def get_company_type(cid):
"""
获取企业填报的企业类型
Parameters:
cid: 企业ID
Returns:
industry
若没有查询到行业 返回空值
"""
# 根据企业ID获取企业当年填报的企业类型
choose_industry = find_rating_inputs({"企业ID": cid, "评价年度": get_current_year()})[0]['行业选择']
if not len(choose_industry) > 1:
return None
else:
return choose_industry[0]
def rating_records_by_cid(cid):
"""
查询一家企业的所有评价记录
Parameters:
cid: 企业ID
Returns:
records: 企业的评价记录
"""
records = find_rating_records({"企业ID": cid})
[record.pop('企业ID') if records is not None else None for record in records]
return records
def start_general_rating_script(cid):
"""
开始进行综合评价的数据填报
Parameters:
cid: 企业ID
Returns:
record 填报信息
"""
# 检查是否已认证
if not check_verified(cid):
return "该企业未认证", {}
# 根据企业ID和当前年度查询是否存在当年的填报记录
exist_input_form = find_rating_inputs({"企业ID": cid, "评价年度": get_current_year()})
# 若存在填报记录根据评价ID查询评价记录该评价是否完成
if exist_input_form:
# 已完成 本年度不能再填报
if find_rating_records({"评价ID": exist_input_form[0]["评价ID"]})[0]['进行状态'] == "完成":
return "本年度已评价", {}
# 未完成 返回已经填报保存的数据
else:
exist_input_form[0].pop('评价ID')
exist_input_form[0].pop('企业ID')
return "继续填报", exist_input_form[0]
# 不存在填报记录 新生成填报记录、评价记录 返回空填报记录对象
else:
# 新生成填报ID
new_rid = gen_new_rid()
# 生成报告期
report_dates = make_3_report_dates()
# 新填报记录
new_input = read_json_file(file_path('/input/static/template/input.json'))
new_input['评价ID'] = new_rid
new_input['企业ID'] = cid
new_input['企业名称'] = find_user_info({"企业ID": cid})[0]['企业名称']
new_input['评价年度'] = get_current_year()
for sheets in ['资产负债表', '利润表', '补充数据表']:
for i in range(3):
new_input['财务填报'][sheets][i]['报告期'] = report_dates[i]
# 生成新评价记录
rating = read_json_file(file_path('/input/static/template/rating.json'))
rating['评价ID'] = new_rid
rating['企业ID'] = cid
rating['评价项目'] = "综合信用评价"
rating['评价方式'] = "企业申报"
rating['进行状态'] = "进行"
# 深拷贝新填报记录不然保存后会生成_id
return_input = copy.deepcopy(new_input)
# 插入新填报记录
insert_general_input(new_input)
# 插入新评价记录
insert_rating_records(rating)
return "开始填报", return_input
def save_general_input_form(cid, input_form):
"""
保存综合信用评价填报
Parameters:
cid: 企业ID
input_form: 填报表单
Returns:
res: desc
"""
rid = find_rating_records({"企业ID": cid, "进行状态": "进行"})[0]['评价ID']
update_general_input(rid, input_form)
def choose_industry_script():
"""
选择行业
Parameters:
-
Returns:
res: desc
"""
data = read_json_file(file_path('/input/static/template/industry.json'))
return data
def questionnaire_script(industry):
"""
问卷模板
Parameters:
industry: 行业
Returns:
data: 问卷内容
"""
data = read_json_file(file_path('/input/static/questionnaire/{}.json'.format(industry)))
return data