1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
# -*- coding: utf-8 -*-
import datetime
import json
import re
from math import ceil, floor
from dateutil.parser import parse
fullformat = lambda dt: dt.strftime('%Y-%m-%d %H:%M:%S')
hourformat = lambda dt: dt.strftime('%H:%M')
def compute_remaining(start, end):
if not isinstance(start, datetime.datetime):
raise ValueError("start needs to be a datetime.datetime")
if not isinstance(end, datetime.datetime):
raise ValueError("start needs to be a datetime.datetime")
seconds = (end - start).total_seconds()
return datetime.timedelta(minutes=floor(seconds / 60)).total_seconds()
class QueryApi(object):
def __init__(self):
pass
def call(self):
raise NotImplementedError("The inheriting class needs to implement "
"the call() method!")
class Departure(object):
def __init__(self, start, end, when, line, since=None):
if since is None:
self.now = datetime.datetime.now()
else:
self.now = since
self.start = start
self.end = end
self.line = line
if isinstance(when, (int, float)):
# We assume to get a UNIX / POSIX timestamp
self.when = datetime.datetime.fromtimestamp(when)
elif isinstance(when, str):
self.when = parse(re.sub('[\s*]$', '', when))
#if (self.when - self.now).total_seconds() < -60:
# self.when = self.when + timedelta(days=1)
elif isinstance(when, datetime.datetime):
self.when = when
else:
raise TypeError("when must be a valid datetime, timestamp or "
"string!")
def __str__(self):
return "Start: %s, End: %s, when: %s, now: %s, line: %s" % (
self.start, self.end, hourformat(self.when),
hourformat(self.now), self.line)
@property
def remaining(self):
return compute_remaining(self.now, self.when)
@property
def to_json(self):
return json.dumps({'start': self.start,
'end': self.end,
'line': self.line,
'now_full': fullformat(self.now),
'now_hour': hourformat(self.now),
'when_full': fullformat(self.when),
'when_hour': hourformat(self.when),
'remaining': round(self.remaining)})
|
