Merge pull request #77 from cowprotocol/use_utc_timezone

Use utc timezone
cowprotocol · Oct 14, 2024 · 7dd2e55 · 7dd2e55
2 parents eab8179 + e042d06
commit 7dd2e55
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/helpers/database.py b/src/helpers/database.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timezone
 
 from hexbytes import HexBytes
 import psycopg
@@ -142,7 +142,9 @@ def write_transaction_timestamp(
             query,
             {
                 "tx_hash": bytes.fromhex(transaction_timestamp[0][2:]),
-                "time": datetime.fromtimestamp(transaction_timestamp[1]),
+                "time": datetime.fromtimestamp(
+                    transaction_timestamp[1], tz=timezone.utc
+                ),
             },
         )
 
@@ -175,7 +177,7 @@ def write_prices_new(self, prices: list[tuple[str, int, float, str]]) -> None:
                     query,
                     {
                         "token_address": bytes.fromhex(token_address[2:]),
-                        "time": datetime.fromtimestamp(time),
+                        "time": datetime.fromtimestamp(time, tz=timezone.utc),
                         "price": price,
                         "source": source,
                     },

diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timezone
 
 from hexbytes import HexBytes
 from sqlalchemy import create_engine, text
@@ -31,7 +31,7 @@ def tests_write_transaction_timestamp():
         "0x" + bytes(res[0]).hex()
         == "0xb75e03b63d4f06c56549effd503e1e37f3ccfc3c00e6985a5aacc9b0534d7c5c"
     )
-    assert res[1].timestamp() == 1728044411
+    assert res[1].replace(tzinfo=timezone.utc).timestamp() == 1728044411
 
 
 def tests_write_transaction_tokens():