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_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