blob: e06ca8913506edb75aaf518903aa0a83aa4afa4e (
plain)
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
|
from datetime import datetime
import pygeoip
ISO_TIMESTAMP = "%Y-%m-%dT%H:%M:%SZ"
class Ban:
def __init__(self, geoip_looker, user=None, timestamp=None, expiry=None, timestamp_format=ISO_TIMESTAMP):
self.geoip_looker = geoip_looker
self.timestamp_format = timestamp_format
self.user = user
self.timestamp = timestamp
self.expiry = expiry
def items(self):
return {
"user": self.user,
"timestamp": self.timestamp.strftime(ISO_TIMESTAMP),
"expiry": self.expiry.strftime(ISO_TIMESTAMP),
}
def hydrate(self, ban_dict):
self.user = ban_dict["user"]
self.timestamp = datetime.strptime(ban_dict["timestamp"], ISO_TIMESTAMP)
self.expiry = datetime.strptime(ban_dict["expiry"], ISO_TIMESTAMP)
def calc_duration(self):
return (self.expiry - self.timestamp).days
def lookup_country_code(self):
try:
return self.geoip_looker.country_code_by_addr(self.user).lower()
except pygeoip.GeoIPError:
return "UNKNOWN"
|