accept string as timestamp

Author: lovvskillz

discord_webhook/webhook.py

@@ -1,7 +1,7 @@
 import json
 import logging
 import time
-from datetime import datetime
+from datetime import datetime, UTC
 from functools import partial
 from http.client import HTTPException
 from typing import Any, Dict, List, Optional, Tuple, Union
@@ -47,7 +47,7 @@ def __init__(
         :keyword dict image: image that will be displayed in the embed
         :keyword dict provider: information about the provider
         :keyword dict thumbnail: thumbnail that will be displayed in the embed
-        :keyword str timestamp: timestamp that will be displayed in the embed
+        :keyword float, int, str, datetime timestamp: timestamp that will be displayed in the embed
         :keyword str title: title of embed
         :keyword str url: add an url to make your embedded title a clickable link
         :keyword dict video: video that will be displayed in the embed
@@ -88,18 +88,21 @@ def set_url(self, url: str) -> None:
         self.url = url
 
     def set_timestamp(
-        self, timestamp: Optional[Union[float, int, datetime]] = None
+        self, timestamp: Optional[Union[float, int, str, datetime]] = None
     ) -> None:
         """
         Set timestamp of the embed content.
-        :param timestamp: timestamp of embed content
+        :param timestamp: optional timestamp of embed content
         """
         if timestamp is None:
-            timestamp = datetime.utcnow()
+            timestamp = datetime.now(UTC)
         elif isinstance(timestamp, float) or isinstance(timestamp, int):
-            timestamp = datetime.utcfromtimestamp(timestamp)
+            timestamp = datetime.fromtimestamp(timestamp, UTC).replace(tzinfo=None)
 
-        self.timestamp = timestamp.isoformat()
+        if not isinstance(timestamp, str):
+            timestamp = timestamp.isoformat()
+
+        self.timestamp = timestamp
 
     def set_color(self, color: Union[str, int]) -> None:
         """

tests/test_embed.py

@@ -42,6 +42,7 @@ def test__set_embed__url(embed):
         1679610926.0,
         datetime.fromisoformat("2023-03-23T22:35:26"),
         datetime.fromisoformat("2023-03-23T23:35:26+01:00"),
+        "2023-03-23T22:35:26",
     ],
 )
 def test__set_embed__timestamp(embed, timestamp):