api-datamanager/Company/CompanyUtils.py

130 lines
3.7 KiB
Python

import xlrd
from DBHelper.MongoHelperInstance import DB_TEST
from Utils.CommonUtil import CommonUtils
class CompanyUtils(object):
@staticmethod
def make_new_cid():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C3.1_工商信息",
{"企业ID": new_cid},
"企业ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_change_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C3.6_工商变更",
{"变更ID": new_cid},
"变更ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_land_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C10.3_土地抵押",
{"抵押ID": new_cid},
"抵押ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_abnormal_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C11.1_经营异常",
{"异常ID": new_cid},
"异常ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_penalties_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C11.2_行政处罚",
{"处罚ID": new_cid},
"处罚ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_violation_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C11.3_严重违法",
{"违法ID": new_cid},
"违法ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_tax_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C11.4_税收违法",
{"违法ID": new_cid},
"违法ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def make_new_arrears_id():
new_cid = CommonUtils.random_code(8)
case = DB_TEST.find_single_column(
"WR_DataBase_v2_test",
"C11.6_欠税公告",
{"欠税ID": new_cid},
"欠税ID"
) is not None
while case:
new_cid = CommonUtils.random_code(8)
return new_cid
@staticmethod
def excel_sheet_parser(**kwargs):
file = kwargs['file']
work_book = xlrd.open_workbook(file_contents=file.read())
sheet = work_book.sheet_by_index(0)
length = len(sheet.col_values(0))
retur_list = list()
if length > 1:
for item in range(1, length):
key_list = sheet.row_values(0)
return_dict = dict()
for num in range(len(key_list)):
return_dict[key_list[num]] = sheet.row_values(item)[num]
retur_list.append(return_dict)
return retur_list