def content(user, file_like, report_run_id):
sess = None
try:
sess = Session()
running_name, finished_name = chellow.dloads.make_names(
FNAME + ".csv", user)
f = open(running_name, mode="w", newline="")
writer = csv.writer(f, lineterminator="\n")
report_run = ReportRun.get_by_id(sess, report_run_id)
props = Contract.get_non_core_by_name(
sess, "configuration").make_properties()
_process_sites(sess, file_like, writer, props, report_run)
report_run.update("finished")
sess.commit()
except BaseException:
msg = traceback.format_exc()
if report_run is not None:
report_run.update("interrupted")
report_run.insert_row(sess, "", ["error"], {"error": msg}, {})
sess.commit()
sys.stderr.write(msg)
writer.writerow([msg])
finally:
if sess is not None:
sess.close()
if f is not None:
f.close()
os.rename(running_name, finished_name)
def content(user, show_ignored, report_run_id):
sess = f = None
try:
sess = Session()
running_name, finished_name = chellow.dloads.make_names(f"{FNAME}.csv", user)
f = open(running_name, mode="w", newline="")
report_run = ReportRun.get_by_id(sess, report_run_id)
props = Contract.get_non_core_by_name(sess, "configuration").make_properties()
ECOES_KEY = "ecoes"
try:
ecoes_props = props[ECOES_KEY]
except KeyError:
raise BadRequest(
f"The property {ECOES_KEY} cannot be found in the configuration "
f"properties."
)
for key in (
"user_name",
"password",
"prefix",
"exclude_mpan_cores",
"ignore_mpan_cores_msn",
):
try:
ecoes_props[key]
except KeyError:
raise BadRequest(
f"The property {key} cannot be found in the 'ecoes' section of "
f"the configuration properties."
)
exclude_mpan_cores = ecoes_props["exclude_mpan_cores"]
ignore_mpan_cores_msn = ecoes_props["ignore_mpan_cores_msn"]
url_prefix = ecoes_props["prefix"]
proxies = props.get("proxies", {})
s = requests.Session()
s.verify = False
r = s.get(url_prefix, proxies=proxies)
data = {
"Username": ecoes_props["user_name"],
"Password": ecoes_props["password"],
}
r = s.post(url_prefix, data=data, allow_redirects=False)
r = s.get(
f"{url_prefix}NonDomesticCustomer/ExportPortfolioMPANs?fileType=csv",
proxies=proxies,
)
_process(
sess,
r.text.splitlines(True),
exclude_mpan_cores,
ignore_mpan_cores_msn,
f,
show_ignored,
report_run,
)
report_run.update("finished")
sess.commit()
except BaseException as e:
msg = traceback.format_exc()
if report_run is not None:
report_run.update("interrupted")
report_run.insert_row(sess, "", ["error"], {"error": msg}, {})
sess.commit()
sys.stderr.write(msg)
f.write(msg)
raise e
finally:
if sess is not None:
sess.close()
if f is not None:
f.close()
os.rename(running_name, finished_name)