def test_read_integer(self):
from os import remove
from utility.file import Read
test_file = 'test_file.txt'
# test single item
string = '35'
with open(test_file, 'w') as file:
file.write(string)
expected = 35
result = Read(test_file).integer()
self.assertEqual(result, expected)
# test multi items
string = '0,1,2,3,4,5'
with open(test_file, 'w') as file:
file.write(string)
expected = [0, 1, 2, 3, 4, 5]
result = Read(test_file).integer()
self.assertListEqual(result, expected)
remove(test_file)
def test_read_decimal(self):
from os import remove
from utility.file import Read
test_file = 'test_file.txt'
# test single item
string = '3.5'
with open(test_file, 'w') as file:
file.write(string)
expected = 3.5
result = Read(test_file).decimal()
self.assertEqual(result, expected)
# test multi items
string = '2.3,4.7,3.6,5.9'
with open(test_file, 'w') as file:
file.write(string)
expected = [2.3, 4.7, 3.6, 5.9]
result = Read(test_file).decimal()
self.assertListEqual(result, expected)
remove(test_file)
def remove_id_from_file(self):
import os
id_file = self.file_list()['completed_ids']
if os.path.exists(id_file):
from utility.file import Read, write
# load delivery ids list
delivery_ids = Read(id_file).integer_list()
# remove id from list
if self.id in delivery_ids:
delivery_ids.pop(delivery_ids.index(self.id))
# update ids file
write(','.join([str(id) for id in delivery_ids]), id_file)
def remove_id_from_file(self):
import os
id_file = self.file_list()['completed_ids']
if os.path.exists(id_file):
from utility.file import Read, write
# load order ids list
id_list = Read(id_file).integer()
# remove id from list
if self.id in id_list:
id_list.pop(id_list.index(self.id))
# update file without id
write(','.join([str(id) for id in id_list]), id_file)
def load_delivery(delivery):
from objects import Delivery
if not isinstance(delivery, Delivery):
# todo: need to write error message
raise TypeError
from objects import Extra_Stop, Order
from os import path
from utility.file import Read
from utility.utility import To_Datetime
# get list of files and directory
file_list = delivery.file_list()
# in progress
if delivery.in_progress:
# start time
if path.exists(file_list['start_time']):
delivery.start_time = Read(file_list['start_time']).datetime()
# distance
if path.exists(file_list['distance']):
delivery.distance = Read(file_list['distance']).decimal()
# average speed
if path.exists(file_list['average_speed']):
delivery.average_speed = Read(file_list['average_speed']).integer()
# end time
if path.exists(file_list['end_time']):
delivery.end_time = Read(file_list['end_time']).datetime()
# completed
else:
# delivery info
delivery_data = Read(file_list['info']).comma()
# distance
if delivery_data[0] != '':
delivery.distance = float(delivery_data[0])
# average speed
if delivery_data[1] != '':
delivery.average_speed = int(delivery_data[1])
# start time
if delivery_data[2] != '':
delivery.start_time = To_Datetime(delivery_data[2]).from_datetime()
# end time
if delivery_data[3] != '':
delivery.end_time = To_Datetime(delivery_data[3]).from_datetime()
return delivery
def load_tip(file_path):
if not isinstance(file_path, str):
raise TypeError
from objects import Tip
from utility.file import Read
data = Read(file_path).decimal()
return Tip(data[0], data[1], data[2])
def assign_id(self):
from os import path
self.id = 0
if path.exists(self.file_list()['running_id']):
from utility.file import Read
self.id = Read(self.file_list()['running_id']).integer()
return self
def load_order(order):
from objects import Order
if not isinstance(order, Order):
# todo: need to fix error message for taking delivery
from resources.error_messages import load__order__wrong_parameter
raise TypeError(load__order__wrong_parameter)
from os import path
from utility.file import Read
from utility.utility import To_Datetime
file_list = order.file_list()
# in progress
if order.in_progress:
# id
if path.exists(file_list['id']):
order.id = Read(file_list['id']).integer()
# tip
if path.exists(file_list['tip']):
order.tip = load_tip(file_list['tip'])
# distance
if path.exists(file_list['distance']):
order.distance = Read(file_list['distance']).decimal()
# end time
if path.exists(file_list['end_time']):
order.end_time = Read(file_list['end_time']).datetime()
# completed
else:
from objects import Tip
# order data
order_data = Read(order.file_list()['info']).comma()
# tip
order.tip = Tip(order_data[0], order_data[1], order_data[2])
# distance
if order_data[3] != '':
order.distance = float(order_data[3])
# end time
if order_data[4] != '':
order.end_time = To_Datetime(order_data[4]).from_datetime()
return order
def load_carry_out_tips(shift):
from objects import Tip
from os import path
from utility.file import Read
if path.exists(shift.file_list()['tips']):
file_data = Read(shift.file_list()['tips']).newline()
for tip in file_data:
tip_data = tip.split(',')
shift.carry_out_tips.append(
Tip(tip_data[0], tip_data[1], tip_data[2]))
return shift
def change_id(self):
# todo: need to write unittest for this
import os
from utility.file import Read, write
# get old id
original_order_id = self.id
# remove file
completed = False
if os.path.exists(self.file_list()['info']):
os.remove(self.file_list()['info'])
completed = True
elif os.path.exists(self.file_list()['id']):
os.remove(self.file_list()['id'])
# user input new id
print(f'\nCurrent I.D. #:{self.id}')
self.input_id()
# update completed ids file
if os.path.exists(self.file_list()['completed_ids']):
# get order id(s) currently in file
order_ids = Read(self.file_list()['completed_ids']).integer_list()
# check if old order id is in list, if so remove it
if original_order_id in order_ids:
order_ids[order_ids.index(original_order_id)] = self.id
# write updated list to completed ids file
write(','.join([str(id) for id in order_ids]),
self.file_list()['completed_ids'])
# rewrite file
if completed is True:
write(self.csv(), self.file_list()['info'])
else:
write(self.id, self.file_list()['id'])
return original_order_id
def load_delivery_orders(delivery):
from objects import Order
from os import path
from utility.file import Read
order_ids_file = Order(delivery).file_list()['completed_ids']
if path.exists(order_ids_file):
# ids
delivery.order_ids = Read(order_ids_file).integer_list()
# orders
delivery.orders =\
[Order(delivery, id).load_completed() for id in delivery.order_ids]
return delivery
def load_shift_deliveries(shift):
from objects import Delivery
from os import path
from utility.file import Read
deliveries_ids_file = Delivery(shift).file_list()['completed_ids']
if path.exists(deliveries_ids_file):
# ids
shift.delivery_ids = Read(deliveries_ids_file).integer_list()
# deliveries
shift.deliveries =\
[Delivery(shift, id).load_completed() for id in shift.delivery_ids]
return shift
def load_split(split):
from objects import Split
if not isinstance(split, Split):
raise TypeError
from os import path
from utility.file import Read
from utility.utility import To_Datetime
file_list = split.file_list()
# in progress
if split.in_progress:
# distance
if path.exists(file_list['distance']):
split.distance = Read(file_list['distance']).decimal()
# start time
if path.exists(file_list['start_time']):
split.start_time = Read(file_list['start_time']).datetime()
# end time
if path.exists(file_list['end_time']):
split.end_time = Read(file_list['end_time']).datetime()
# completed
else:
# split info
split_info = Read(file_list['info']).comma()
# distance
if split_info[0] != '':
split.distance = float(split_info[0])
# start time
if split_info[1] != '':
split.start_time = To_Datetime(split_info[1]).from_datetime()
# end time
if split_info[2] != '':
split.end_time = To_Datetime(split_info[2]).from_datetime()
return split
def remove_id_from_file(self):
from utility.file import Read, write
from utility.utility import To_Datetime
ids_file = self.file_list()['completed_ids']
id_list = [
To_Datetime(id).from_date() for id in Read(ids_file).comma()
]
new_ids_list = [str(id.date()) for id in id_list if id != self.id]
if len(new_ids_list) == 0:
from os import remove
remove(ids_file)
elif len(new_ids_list) > 0:
from utility.file import write
write(','.join(new_ids_list), ids_file)
def load_parent_extra_stops(parent):
from objects import Delivery, Shift
if not isinstance(parent, (Delivery, Shift)):
raise TypeError
from objects import Extra_Stop
from os import path
from utility.file import Read
extra_stop_ids_file = Extra_Stop(parent).file_list()['completed_ids']
if path.exists(extra_stop_ids_file):
# ids
parent.extra_stop_ids = Read(extra_stop_ids_file).integer_list()
# extra stops
parent.extra_stops = [
Extra_Stop(parent, id).load_completed()
for id in parent.extra_stop_ids
]
return parent
from objects import Shift
import os
from utility.file import Read
from utility.utility import now, To_Datetime
id_list = Read(Shift(now().date()).file_list()['completed_ids']).comma()
uncompleted_id_list = []
os.chdir(os.path.join('data', 'shifts'))
for directory in os.listdir():
if directory not in id_list:
uncompleted_id_list.append(directory)
print(f'Total uncompleted shifts: {len(uncompleted_id_list)}')
print('List of uncompleted dates:')
for directory in uncompleted_id_list:
print(directory)
def load_extra_stop(extra_stop):
from objects import Extra_Stop
if not isinstance(extra_stop, Extra_Stop):
raise TypeError
from objects import Delivery, Shift
from os import path
from utility.file import Read
from utility.utility import To_Datetime
# get files and directory
file_list = extra_stop.file_list()
# in progress
if extra_stop.in_progress:
# location
if path.exists(file_list['location']):
extra_stop.location = Read(file_list['location']).data
# reason
if path.exists(file_list['reason']):
extra_stop.reason = Read(file_list['reason']).data
# distance
if path.exists(file_list['distance']):
extra_stop.distance =\
Read(file_list['distance']).decimal()
# end time
if path.exists(file_list['end_time']):
extra_stop.end_time = Read(file_list['end_time']).datetime()
if isinstance(extra_stop.parent, Shift):
# start time
if path.exists(file_list['start_time']):
extra_stop.start_time = Read(
file_list['start_time']).datetime()
# completed
else:
# extra stop info
extra_stop_data = Read(file_list['info']).newline()
# location
if extra_stop_data[0] != '':
extra_stop.location = extra_stop_data[0]
# reason
if extra_stop_data[1] != '':
extra_stop.reason = extra_stop_data[1]
# distance
if extra_stop_data[2] != '':
extra_stop.distance = float(extra_stop_data[2])
if isinstance(extra_stop.parent, Delivery):
# end time
if extra_stop_data[3] != '':
extra_stop.end_time = To_Datetime(
extra_stop_data[3]).from_datetime()
elif isinstance(extra_stop.parent, Shift):
# start time
if extra_stop_data[3] != '':
extra_stop.start_time =\
To_Datetime(extra_stop_data[3]).from_datetime()
# end time
if extra_stop_data[4] != '':
extra_stop.end_time =\
To_Datetime(extra_stop_data[4]).from_datetime()
return extra_stop
def load_shift(shift):
from objects import Shift
if not isinstance(shift, Shift):
raise TypeError
from objects import Delivery, Extra_Stop, Split
from os import path
from utility.file import Read
from utility.utility import To_Datetime
file_list = shift.file_list()
# in progress
if shift.in_progress:
# start time
if path.exists(file_list['start_time']):
shift.start_time = Read(file_list['start_time']).datetime()
# end time
if path.exists(file_list['end_time']):
shift.end_time = Read(file_list['end_time']).datetime()
# distance
if path.exists(file_list['distance']):
shift.distance = Read(file_list['distance']).decimal()
# fuel economy
if path.exists(file_list['fuel_economy']):
shift.fuel_economy = Read(file_list['fuel_economy']).decimal()
# vehical compensation
if path.exists(file_list['vehicle_compensation']):
shift.vehicle_compensation =\
Read(file_list['vehicle_compensation']).decimal()
# device compensation
if path.exists(file_list['device_compensation']):
shift.device_compensation =\
Read(file_list['device_compensation']).decimal()
# total hours
if path.exists(file_list['total_hours']):
shift.total_hours = Read(file_list['total_hours']).decimal()
# extra tips claimed
if path.exists(file_list['extra_tips_claimed']):
shift.extra_tips_claimed =\
Read(file_list['extra_tips_claimed']).decimal()
# completed
else:
# shift info
shift_data = Read(file_list['info']).comma()
# distance
if shift_data[0] != '':
shift.distance = float(shift_data[0])
# fuel economy
if shift_data[1] != '':
shift.fuel_economy = float(shift_data[1])
# vehicle compensation
if shift_data[2] != '':
shift.vehicle_compensation = float(shift_data[2])
# device compensation
if shift_data[3] != '':
shift.device_compensation = float(shift_data[3])
# extra tips claimed
if shift_data[4] != '':
shift.extra_tips_claimed = float(shift_data[4])
# total hours
if shift_data[5] != '':
shift.total_hours = float(shift_data[5])
# start time
if shift_data[6] != '':
shift.start_time = To_Datetime(shift_data[6]).from_datetime()
# end time
if shift_data[7] != '':
shift.end_time = To_Datetime(shift_data[7]).from_datetime()
return shift