Put a helper function in utils to do iso8601 formatting. change IRCServer's

last-read setting to use it.

2 files changed, 7 insertions, 3 deletions

diff --git a/src/utils/__init__.py b/src/utils/__init__.py
index 692d305a..9cb99e93 100644
--- a/src/utils/__init__.py
+++ b/src/utils/__init__.py
@@ -1,7 +1,11 @@
-import decimal, io, ipaddress, re, typing
+import datetime, decimal, io, ipaddress, re, typing
 from src.utils import cli, consts, irc, http, parse, security
 
-ISO8601_FORMAT = "%Y-%m-%dT%H:%M:%S%z"
+ISO8601_PARSE = "%Y-%m-%dT%H:%M:%S%z"
+
+def iso8601_format(dt: datetime.datetime) -> str:
+    formatted = dt.isoformat(timespec="milliseconds")
+    return "%sZ" % formatted
 
 TIME_SECOND = 1
 TIME_MINUTE = TIME_SECOND*60
diff --git a/src/utils/cli.py b/src/utils/cli.py
index 598bbc2a..638ff749 100644
--- a/src/utils/cli.py
+++ b/src/utils/cli.py
@@ -4,7 +4,7 @@ def bool_input(s: str):
     result = input("%s (Y/n): " % s)
     return not result or result[0].lower() in ["", "y"]
 
-def add_server(database: Database.Database):
+def add_server(database: "Database.Database"):
     alias = input("alias: ")
     hostname = input("hostname: ")
     port = int(input("port: "))