import csv from datetime import datetime import sys import os from discord_webhook import DiscordWebhook webhook = DiscordWebhook( url='https://discordapp.com/api/webhooks/744608404848246926/FwwWUNmuWsCetFDBOcA1trCx6nSflI0yX3f9BNh7-oB5a2VYSLl882di23XK2TNWwQxP', username="billing tmgdsp") dirpath = "." FILE = [filename for filename in os.listdir(dirpath) if filename.startswith("TMGDSP_")] FILE = FILE[0] ext_date_from_filaname = FILE.split('_')[-1] print(ext_date_from_filaname) #usrmonth = input("Month to be processed: ") countofnotsms = int(1) dictionary = {} def add_dict(id, key, value): dict = dictionary.get(id) if key in dict: print("increasing notsms dict key for: ",value,"key IS: ",key) dict[key] += value print("end state: ",dict[key]) dict[key] = round(dict[key],5) print("key: ",key) else: print("Never should happen", id) def add_dictsms(id, pricesms): dict = dictionary.get(id) print("increasing sms dict key for dict: ",dict," for :",pricesms) dict["smscount"] += pricesms def add_dictdataamount(id, dataamount): dict = dictionary.get(id) print("increasing dataamount dict key for dict: ",dict," for :",dataamount) dict["dataamount"] += int(dataamount)/1024 #MAIN with open(FILE, newline='', encoding='utf-8') as csvfile: reader = csv.DictReader(csvfile, delimiter=';') for row in reader: #print(row) if row['MSISDN'] == '': continue #skip empty MSISDN #if row['BILLED_PERIOD_START'] == '': #continue #my_string = row['BILLED_PERIOD_START'] #my_string = '2021-09-06' #print(my_string) #try: #my_date = datetime.strptime(my_string, "%Y-%m-%d") #except: #print("An exception occurred, current my_string is: ", my_string) #sys.exit(1) #print(row['BILLED_PERIOD_START']) #print(my_date.month) #if my_date.month != '5': #continue #skip bad dates MSISDN = int(row['MSISDN']) ICCID = int(row['ICCID']) IMSI = int(row['IMSI']) print("charge_net: ",row['CHARGE_NET']) price = float(row['CHARGE_NET'].replace(',','.')) print("before conditions price: ",price, "for: ",ICCID) typ = None if row['CHARGE_TYPE'] == "Pravidelné měsíční poplatky": typ = "data" elif row['CHARGE_TYPE'] == "Poplatky za roamingový datový provoz": typ = "data" elif row['CHARGE_TYPE'] == "Poplatky za národní datový provoz": typ = "data" elif row['CHARGE_TYPE'] == "Poplatky za roamingový SMS provoz": typ = "sms" elif row['CHARGE_TYPE'] == "Poplatky za národní SMS provoz": typ = "sms" else: typ = 'other' if row['INCREMENT'] == "SMS": pricesms = int(row['RATED_AMOUNT']) else: pricesms = 0 #only if not exist create new record in dictionary for ID if not MSISDN in dictionary: dictionary[MSISDN] = {'IMSI':IMSI, 'ICCID':ICCID,'data':0.00000, 'sms':0.00000, 'other':0.00000, 'dataamount':0, 'smscount':pricesms} print("Creating new dictionary: ", dictionary[MSISDN]) if dictionary[MSISDN] == "436761890300016": print("creating dictionary for our unit") input() print("Adding: ", price, " for: ", dictionary[MSISDN], "TYP: ",typ) add_dict(MSISDN, typ, price) if MSISDN == 436761890301205: print("created dictionary for our unit") if row['INCREMENT'] == "SMS": print("pricesms: ",pricesms," for unit: ",dictionary[MSISDN]) add_dictsms(MSISDN, pricesms) if MSISDN == 436761890301205: print("increasing pricesms") if row['INCREMENT'] == "1/1 kB": if row['RATED_AMOUNT'] != "0": dataamount = row['RATED_AMOUNT'] print(dataamount) add_dictdataamount(MSISDN, dataamount) else: dataamount = 0 #ptry: #print(dictionary[436761890319959]) #one with both types #pexcept: #pprint("Required unit is not in our set") #psys.exit(1) #pprint(len(dictionary.keys())) #for key, value in dictionary.items(): #print(key, ' : ', value) print("Count of units: ", len(dictionary.keys())) #print("Count of sms: ",countofsms) #for dict_value in dictionary: # type(dict_value) # #for k, v in dict_value.items(): #with open('fakturace_tmgdsp_export_20211110v2.csv', 'w') as f: # for key in dictionary.keys(): # print(key,dictionary[key]) # f.write("%s,%s\n"%(key,dictionary[key])) with open('output_TMGDSP_'+ext_date_from_filaname, 'w') as f: writer = csv.writer(f) writer.writerow(["MSISDN","ICCID","IMSI","DATA-PRICE","DATA-AMOUNT","SMS-PRICE","SMS-COUNT","OTHER-PRICE","SIM-STATUS", "OPERATOR"]) for key in dictionary.keys(): # Getting all required variables to write them into export MSISDN = key ICCID = dictionary[key]['ICCID'] IMSI = dictionary[key]['IMSI'] data_price = dictionary[key]['data'] sms_price = dictionary[key]['sms'] other_price = dictionary[key]['other'] sms_count = dictionary[key]['smscount'] data_amount = dictionary[key]['dataamount'] print("MSISDN: ", MSISDN) print("ICCID: ", ICCID) print("IMSI: ", IMSI) print("data_price: ", data_price) print("sms_price: ", sms_price) print("sms_count: ", sms_count) writer.writerow([MSISDN, ICCID, IMSI, data_price, data_amount, sms_price, sms_count, other_price, "N/A", "TMGDSP"]) csvfile.close() with open('output_TMGDSP_'+ext_date_from_filaname, "rb") as f: webhook.add_file(file=f.read(), filename='output_TMGDSP_'+ext_date_from_filaname) response = webhook.execute() #csv_columns = ['takhleuztoneudelam', 'ICCID', 'data', 'sms', 'other', 'smscount'] #try: # with open('output_tmgdsp_202112.csv', 'w') as csvfile: # w = csv.DictWriter(sys.stdout, csv_columns) # for key, val in sorted(dictionary.items()): #row = {'takhleuztoneudelam': key} #row.update(val) #writer = csv.DictWriter(csvfile, fieldnames=csv_columns) #writer.writerow(row) #except IOError: #print(f"I/O error: {IOError}") #csvfile.close()