#!/usr/bin/python
import csv
import os
#---------------------------------------------------------------------+
# GET START AND END DATE FROM INPUT FILE
#---------------------------------------------------------------------+
with open('/home/sindh/www-htdocs/pakistan_dss/uploads/user/DB/Tori_Model/indus.csv') as file:
    first_line = file.readline()
    for last_line in file:
        pass
file.close()
data_first = first_line.strip()
data_arr_first = data_first.split(',')
start_dt = data_arr_first[0]
#print(start_dt)
data_last = last_line.strip()
data_arr_last = data_last.split(',')
end_dt = data_arr_last[0]
#print(end_dt)
#---------------------------------------------------------------------+
# GET BC (CHENAB), PUT DATA IN ARRAY
#---------------------------------------------------------------------+
bc_c=[]
with open('/home/sindh/www-htdocs/pakistan_dss/uploads/user/DB/Tori_Model/chenab.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    count_c = 0
    hour = 0
    for row in csv_reader:
        count_c = count_c+1
        tp_c = round(float(row[1]),0)
        bc = (str(hour).rjust(4," "))+(str(tp_c).rjust(8,"0"))
        bc_c.append(bc)
        hour = hour + 24
csv_file.close() 
#print(bc_c)
#---------------------------------------------------------------------+
# GET BC (INDUS), PUT DATA IN ARRAY
#---------------------------------------------------------------------+
bc_i=[]
with open('/home/sindh/www-htdocs/pakistan_dss/uploads/user/DB/Tori_Model/indus.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    count_i = 0
    hour = 0
    for row_ in csv_reader:
        count_i = count_i+1
        tp = round(float(row_[1]),0)
        bc = (str(hour).rjust(4," "))+(str(tp).rjust(8,"0"))
        bc_i.append(bc)
        hour = hour + 24
csv_file.close()
#---------------------------------------------------------------------+
# UPDATE MODEL .B01 FILE
#---------------------------------------------------------------------+
no_indus=1
no_chenab=1
file_path = '/home/sindh/www-htdocs/pakistan_dss/DATA/ML/MODELS/Tori_Model/Sindh_Complete.b01'
with open(file_path+str('.txt'),"r") as file:
    data= file.readlines()
with open(file_path,'w') as b:
    for i,line in enumerate(data):
        if i == 49:                                                     #Checked
            b.write("  Start Date/Time       = "+start_dt+" 2400\n")
            b.write("  End Date/Time         = "+end_dt+" 2400\n")
        if i == 65:                                                     #Checked
            b.write("    "+ str(count_i) +"\n")        
            a=1
            for idx in range(count_i):
                b.write("    "+bc_i[idx])
                rem = a%5
                if (rem==0):
                    if no_indus == len(bc_i):
                        break
                    b.write("\n")                    
                a = a+1
                no_indus=no_indus+1
            b.write("\n")        
        if i == 68:                                                     #Checked
            b.write("    "+ str(count_c) +"\n")
            a=1
            for idx in range(count_c):
                b.write("    "+bc_c[idx])
                rem = a%5
                if (rem==0):
                    if no_chenab == len(bc_c):
                        break
                    b.write("\n")                    
                a = a+1
                no_chenab=no_chenab+1 
            b.write("\n")
        b.write(line)