156 lines
4.0 KiB
Python
156 lines
4.0 KiB
Python
import csv
|
|
from datetime import datetime
|
|
import sys
|
|
import pandas as pd
|
|
|
|
FILE = 'TMGDSP202110.CSV'
|
|
|
|
#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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#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'])
|
|
|
|
print("charge_net: ",row['CHARGE_NET'])
|
|
price = float(row['CHARGE_NET'].replace(',','.'))
|
|
price = round(price, 5)
|
|
|
|
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] = {'ICCID':ICCID,'data':0.00000, 'sms':0.00000, 'other':0.00000, 'smscount':pricesms}
|
|
print("Creating new dictionary: ", dictionary[MSISDN])
|
|
|
|
|
|
print("Adding: ", price, " for: ", dictionary[MSISDN], "TYP: ",typ)
|
|
add_dict(MSISDN, typ, price)
|
|
if row['INCREMENT'] == "SMS":
|
|
print("pricesms: ",pricesms," for unit: ",dictionary[MSISDN])
|
|
add_dictsms(MSISDN, pricesms)
|
|
|
|
|
|
|
|
|
|
|
|
#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]))
|
|
|
|
|
|
|
|
csv_columns = ['takhleuztoneudelam', 'ICCID', 'data', 'sms', 'other', 'smscount']
|
|
try:
|
|
with open('fakturace_tmgdsp_export_202110v1.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}") |