29 lines
967 B
Python
29 lines
967 B
Python
from datetime import datetime, timezone
|
|
from cognee.modules.engine.models import Interval, Timestamp, Event
|
|
from cognee.modules.engine.utils import generate_node_id
|
|
|
|
|
|
def generate_timestamp_datapoint(ts: Timestamp) -> Timestamp:
|
|
"""Create a Timestamp datapoint from a Timestamp model."""
|
|
time_at = date_to_int(ts)
|
|
timestamp_str = (
|
|
f"{ts.year:04d}-{ts.month:02d}-{ts.day:02d} {ts.hour:02d}:{ts.minute:02d}:{ts.second:02d}"
|
|
)
|
|
return Timestamp(
|
|
id=generate_node_id(str(time_at)),
|
|
time_at=time_at,
|
|
year=ts.year,
|
|
month=ts.month,
|
|
day=ts.day,
|
|
hour=ts.hour,
|
|
minute=ts.minute,
|
|
second=ts.second,
|
|
timestamp_str=timestamp_str,
|
|
)
|
|
|
|
|
|
def date_to_int(ts: Timestamp) -> int:
|
|
"""Convert timestamp to integer milliseconds."""
|
|
dt = datetime(ts.year, ts.month, ts.day, ts.hour, ts.minute, ts.second, tzinfo=timezone.utc)
|
|
time = int(dt.timestamp() * 1000)
|
|
return time
|