You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.8 KiB
49 lines
1.8 KiB
5 years ago
|
import locale
|
||
|
|
import threading
|
||
|
|
|
||
|
|
from re import search
|
||
|
|
from datetime import datetime
|
||
|
|
from contextlib import contextmanager
|
||
|
|
|
||
|
|
class formatter:
|
||
|
|
|
||
|
|
# source setlocale = https://stackoverflow.com/questions/18593661/how-do-i-strftime-a-date-object-in-a-different-locale
|
||
|
|
LOCALE_LOCK = threading.Lock()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
@contextmanager
|
||
|
|
def setlocale(name):
|
||
|
|
with formatter.LOCALE_LOCK:
|
||
|
|
saved = locale.setlocale(locale.LC_ALL)
|
||
|
|
try:
|
||
|
|
yield locale.setlocale(locale.LC_ALL, name)
|
||
|
|
finally:
|
||
|
|
locale.setlocale(locale.LC_ALL, saved)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def date_time_merger(line, invoice_date):
|
||
|
|
with formatter.setlocale('nl_NL.UTF-8'):
|
||
|
|
invoice_datetime = datetime.strptime(invoice_date, '%d %B %Y')
|
||
|
|
invoice_year = invoice_datetime.year - 1 if invoice_datetime.month == 1 else invoice_datetime.year
|
||
|
|
date = [x for x in line if search("\d{2} \w{3}",x)]
|
||
|
|
time = [x for x in line if search("\d{2}:\d{2}",x)]
|
||
|
|
if len(date) < 1 or len(time) < 1:
|
||
|
|
print("Warning: Could not find date and time in {0}".format(line))
|
||
|
|
return line
|
||
|
|
datetime_string = '{0} {1}'.format(date[0], time[0])
|
||
|
|
with formatter.setlocale('nl_NL.UTF-8'):
|
||
|
|
linedate = datetime.strptime(datetime_string, '%d %b %H:%M').replace(year = invoice_year)
|
||
|
|
datestring = linedate.strftime('%Y-%m-%d %H:%M:%S')
|
||
|
|
return [line[0], datestring] +line[3:]
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def remove_prefix(element):
|
||
|
|
element_text = element.text
|
||
|
|
if len(element_text) < 5: return
|
||
|
|
if element_text[3] == ' ':
|
||
|
|
element.text = element_text[4:]
|
||
|
|
return
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def concat_line(line):
|
||
|
|
return ''.join([element + ';' for element in line])[:-1] + '\n'
|