paramecio/parameciofm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixes in datetimetest

Browse source
This commit is contained in:
Antonio de la Rosa 2017-11-21 03:57:32 +01:00
parent bcabada26b
commit ae15a6e358
2 changed files with 181 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

171
paramecio/citoplasma/datetime.py
View file

@ -101,7 +101,7 @@ def checkdatetime(y, m, d, h, mi, s):
# Get the localtime
def now(utc=False):
def now(utc=False, tz=''):
"""
actual=datetime.today()
@ -118,20 +118,22 @@ def now(utc=False):
"""
if not utc:
if tz=='':
actual=arrow.now().format(sql_format_time)
else:
actual=arrow.to(tz).now().format(sql_format_time)
else:
actual=arrow.utcnow().format(sql_format_time)
return actual
def today(utc=False):
def today(utc=False,tz=''):
return now()[:8]+'000000'
return now(utc, tz)[:8]+'000000'
# Get actual timestamp
def obtain_timestamp(timeform, local=False):
def obtain_timestamp(timeform, local=False, tz=''):
y, m, d, h, mi, s=format_timedata(timeform)
@ -145,7 +147,10 @@ def obtain_timestamp(timeform, local=False):
#return timestamp-offset
t=arrow.arrow.Arrow(y, m, d, h, mi, s).to(environ['TZ'])
if tz=='':
tz=environ['TZ']
t=arrow.arrow.Arrow(y, m, d, h, mi, s).to(tz)
timestamp=t.timestamp
@ -170,7 +175,7 @@ def timestamp_to_datetime(timestamp):
# Get a utc timestamp and convert to local
def timestamp_to_datetime_local(timestamp):
def timestamp_to_datetime_local(timestamp, tz=''):
#time_set=time.localtime(timestamp)
@ -178,7 +183,10 @@ def timestamp_to_datetime_local(timestamp):
t=arrow.get(timestamp)
return t.to(environ['TZ']).format(sql_format_time)
if tz=='':
tz=environ['TZ']
return t.to(tz).format(sql_format_time)
def format_datetime(format_time, timeform, func_utc_return):
@ -241,7 +249,7 @@ def format_local_strtime(strtime, timeform):
#Input is utc timestamp, return local arrow object
def sum_utc(timestamp):
def sum_utc(timestamp, tz=''):
#offset=time.altzone
@ -249,11 +257,14 @@ def sum_utc(timestamp):
t=arrow.get(timestamp)
return t.to(environ['TZ'])
if tz=='':
tz=environ['TZ']
return t.to(tz)
#Input is local timestamp, return utc arrow object
def substract_utc(timestamp):
def substract_utc(timestamp, tz=''):
#offset=time.altzone
@ -265,7 +276,10 @@ def substract_utc(timestamp):
y, m, d, h, mi, s=format_timedata(timeform)
t=arrow.get(datetime(y, m, d, h, mi, s), environ['TZ']).to('UTC')
if tz=='':
tz=environ['TZ']
t=arrow.get(datetime(y, m, d, h, mi, s), tz).to('UTC')
return t
@ -276,24 +290,38 @@ def no_utc(timestamp):
class TimeClass:
def __init__(self, timestamp=0, utc=False):
def __init__(self, timestamp=0, tz=''):
self.utc=utc
self.utc=False
self.format_time=sql_format_time
self.format_time_txt=format_time_txt
if timestamp==0:
self.format_date_txt=format_date_txt
self.datetime=now(self.utc)
self.format_date_full=format_date_txt+' '+format_time_txt
self.tz=environ['TZ']
if tz:
self.tz=tz
if type(timestamp).__name__=='int':
if self.utc:
self.datetime=utcnow()
else:
self.datetime=timestamp_to_datetime(timestamp)
else:
self.datetime=timestamp_to_datetime(timestamp)
self.datetime=timestamp
y, m, d, h, mi, s=format_timedata(self.datetime)
self.t=arrow.get(datetime(y, m, d, h, mi, s))
self.t=arrow.get(datetime(y, m, d, h, mi, s), self.tz)
def add_month(self, num_months):
@ -301,7 +329,112 @@ class TimeClass:
return m.format(self.format_time)
def substract_month(self, num_months):
m=self.t.shift(months=-num_months)
return m.format(self.format_time)
def add_day(self, num_days):
m=self.t.shift(days=+num_days)
return m.format(self.format_time)
def substract_day(self, num_days):
m=self.t.shift(days=-num_days)
return m.format(self.format_time)
def add_year(self, num_years):
m=self.t.shift(years=+num_years)
return m.format(self.format_time)
def substract_year(self, num_years):
m=self.t.shift(years=-num_years)
return m.format(self.format_time)
def format(self):
return m.format(self.format_time_txt)
return self.t.format(self.format_date_full)
def local_to_utc(self):
self.t=self.t.to('utc')
# Only use
def utc_to_local(self):
self.t=self.t.to(self.tz)
def local_to_tz(self, tz):
self.t=self.t.to(tz)
def now(self, utc=False):
if not utc:
actual=self.t.now().format(sql_format_time)
else:
actual=self.t.utcnow().format(sql_format_time)
return actual
def today(self, utc=False):
return now(utc, self.tz)[:8]+'000000'
def timestamp_to_datetime(self, timestamp, utc=False):
if utc:
return self.t.get(timestamp).to('UTC').format(sql_format_time)
else:
return self.t.get(timestamp).format(sql_format_time)
def obtain_timestamp(self, timeform, utc=False):
y, m, d, h, mi, s=format_timedata(timeform)
if checkdatetime(y, m, d, h, mi, s):
#timestamp=int(time.mktime((y, m, d, h, mi, s, 0, 0, -1)))
if local:
#offset=time.altzone
#return timestamp-offset
t=arrow.arrow.Arrow(y, m, d, h, mi, s).to(tz)
timestamp=t.timestamp
else:
timestamp=arrow.arrow.Arrow(y, m, d, h, mi, s).timestamp
return timestamp
#return mktime($h, $mi, $s, $m, $d, $y);
else:
return False
def format_strtime(self, strtime, timeform):
timestamp=obtain_timestamp(timeform)
if timestamp:
return self.t.get(timestamp).format(strtime)
else:
return False

26
tests/datetimetest.py
View file

@ -76,5 +76,31 @@ class TestFieldMethods(unittest.TestCase):
print(datetime.local_to_utc('20121221232421', tz))
"""
class TestClassMethods(unittest.TestCase):
def test_timenow(self):
datetime.timezone='Europe/Madrid'
datetime.set_timezone()
d=datetime.TimeClass('20121126231248')
self.assertEqual('2012/11/26 23:12:48', d.format())
d.local_to_utc()
self.assertEqual('2012/11/26 22:12:48', d.format())
d.utc_to_local()
self.assertEqual('20130126231248', d.add_month(2))
self.assertEqual('20120926231248', d.substract_month(2))
self.assertEqual('20121203231248', d.add_day(7))
self.assertEqual('20121119231248', d.substract_day(7))
#self.assertEqual('20121203231248', d.substract_day(7))
if __name__ == '__main__':
unittest.main()