Coverage for adrs/data/utils.py: 36%

1import re 

2 

3from pathlib import Path 

4from datetime import datetime, timezone 

5 

6from adrs.types import Topic 

7 

8DATE_RE = re.compile(r"(\d{4}-\d{2}-\d{2})") 

9 

10 

11def make_filename(topic: Topic) -> str: 

12 return ( 

13 (topic.query_params_str() or "data") 

14 .replace("-", "_") 

15 .replace("|", "_") 

16 .replace("?", "_") 

17 .replace("/", "_") 

18 .replace("=", "_") 

19 .replace("&", "_") 

20 ) 

21 

22 

23def make_filepath(topic: Topic, fmt: str, suffix: str | None = None) -> Path: 

24 filename = make_filename(topic) 

25 if suffix: 

26 filename = f"{filename}_{suffix}" 

27 path = f"{topic.provider}|{topic.endpoint}".replace("coinglass-v4", "coinglass") 

28 transformed = path.replace("/", "_").replace("|", "/") 

29 return Path(transformed) / f"{filename}.{fmt}" 

30 

31 

32def parse_date_from_filename(filename: str) -> datetime | None: 

33 m = DATE_RE.search(filename) 

34 if m: 

35 try: 

36 return datetime.fromisoformat(m.group(1)).replace(tzinfo=timezone.utc) 

37 except ValueError: 

38 return None 

39 return None