Coverage for src/lexigram/graphql/scalars/timestamp.py: 75%

1from __future__ import annotations 

2 

3from datetime import datetime 

4 

5 

6class Timestamp: 

7 """Custom scalar for timestamp (Unix epoch) values.""" 

8 

9 @staticmethod 

10 def serialize(dt: datetime | None) -> int | None: 

11 if dt is None: 

12 return None 

13 if isinstance(dt, datetime): 

14 return int(dt.timestamp()) 

15 raise ValueError(f"Cannot serialize {type(dt)} as Timestamp") 

16 

17 @staticmethod 

18 def parse_value(value: int | None) -> datetime | None: 

19 if value is None: 

20 return None 

21 return datetime.fromtimestamp(value) 

22 

23 

24__all__ = [ 

25 "Timestamp", 

26]