def zip_for_broker(
broker,
period,
include=INCLUDE,
):
start, end = period
files = [
construct_filenames_of_generation(type, broker, period)
for type in include
]
result_zip_name = "{}@{}--{}-{}.zip".format(
broker, start, end, "-".join(include)
)
try:
password = get_password_for_broker(broker)
if password:
return compress(
files,
CONFIG['out']['out_dir'],
result_zip_name,
password
)
return False
except (OSError, IOError) as e:
LOGGER.fatal(
"Falied to create archive {} for broker {}. Error: {}".format(
result_zip_name, broker, e
)
)
return False
def zip_all_bids(brokers, period):
start, end = period
files = ["all@{}--{}-bids.csv".format(start, end)]
files.extend([
"{}@{}--{}-invoices.csv".format(broker, start, end)
for broker in brokers
if broker != 'all'
])
try:
name = "all@{}--{}-bids.zip".format(start, end)
password = get_password_for_broker('all')
if password:
return compress(
files,
CONFIG['out']['out_dir'],
name,
password
)
return False
except (OSError, IOError) as e:
LOGGER.fatal(
"Falied to create archive {} for broker {}. Error: {}".format(
name, broker, e
)
)
return False
def zip_all_bids(brokers, period):
start, end = period
files = ["all@{}--{}-bids.csv".format(start, end)]
files.extend([
"{}@{}--{}-invoices.csv".format(broker, start, end)
for broker in brokers
if broker != 'all'
])
try:
name = "all@{}--{}-bids.zip".format(start, end)
password = get_password_for_broker('all')
if password:
return compress(
files,
CONFIG['out']['out_dir'],
name,
password
)
return False
except (OSError, IOError) as e:
LOGGER.fatal(
"Falied to create archive {} for broker {}. Error: {}".format(
name, broker, e
)
)
return False
def zip_for_broker(
broker,
period,
include=INCLUDE,
):
start, end = period
files = [
construct_filenames_of_generation(type, broker, period)
for type in include
]
result_zip_name = "{}@{}--{}-{}.zip".format(
broker, start, end, "-".join(include)
)
try:
password = get_password_for_broker(broker)
if password:
return compress(
files,
CONFIG['out']['out_dir'],
result_zip_name,
password
)
return False
except (OSError, IOError) as e:
LOGGER.fatal(
"Falied to create archive {} for broker {}. Error: {}".format(
result_zip_name, broker, e
)
)
return False
def zip_all_tenders_prozorro_market(period):
start, end = period
try:
return compress(
["all@{}--{}-tenders_prozorro_market.csv".format(start, end)],
CONFIG['out']['out_dir'],
"all@{}--{}-tenders_prozorro_market.zip".format(start, end),
None
)
except (OSError, IOError) as e:
LOGGER.fatal("Error: {}".format(e))
return False
def zip_all_tenders(brokers, period):
start, end = period
try:
return compress([
"{}@{}--{}-{}.csv".format(broker, start, end, type)
for type in ['tenders', 'refunds']
for broker in brokers if brokers != 'all'
], CONFIG['out']['out_dir'],
"all@{}--{}-tenders.zip".format(start, end), None)
except (OSError, IOError) as e:
LOGGER.fatal("Error: {}".format(e))
return False
def zip_all_tenders(brokers, period):
start, end = period
try:
return compress(
[
"{}@{}--{}-{}.csv".format(broker, start, end, type)
for type in ['tenders', 'refunds']
for broker in brokers
if brokers != 'all'
],
CONFIG['out']['out_dir'],
"all@{}--{}-tenders.zip".format(start, end),
None
)
except (OSError, IOError) as e:
LOGGER.fatal("Error: {}".format(e))
return False