pybgpkitstream 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

pybgpkitstream/bgpkitstream.py

@@ -56,6 +56,15 @@ class Directory:
         pass
 
 
+def get_shared_memory():
+    """Get a RAM-based temp path if available, otherwise fall back to default."""
+    if os.path.exists("/dev/shm"):  # Linux tmpfs
+        return "/dev/shm"
+    elif os.path.exists("/Volumes/RAMDisk"):  # macOS (if mounted)
+        return "/Volumes/RAMDisk"
+    return None  # Fall back to default temp directory
+
+
 class BGPKITStream:
     def __init__(
         self,
@@ -73,7 +82,9 @@ class BGPKITStream:
         self.collector_id = collector_id
         self.data_type = data_type
         self.cache_dir: Directory | TemporaryDirectory = (
-            Directory(cache_dir) if cache_dir else TemporaryDirectory()
+            Directory(cache_dir)
+            if cache_dir
+            else TemporaryDirectory(dir=get_shared_memory())
         )
         self.filters = filters
         self.max_concurrent_downloads = max_concurrent_downloads
@@ -116,9 +127,8 @@ class BGPKITStream:
         self.urls = {"rib": defaultdict(list), "update": defaultdict(list)}
         for data_type in self.data_type:
             items: list[BrokerItem] = self.broker.query(
-                ts_start=datetime.datetime.fromtimestamp(self.ts_start)
-                - datetime.timedelta(minutes=1),
-                ts_end=datetime.datetime.fromtimestamp(self.ts_end),
+                ts_start=int(self.ts_start - 60),
+                ts_end=int(self.ts_end),
                 collector_id=self.collector_id,
                 data_type=data_type,
             )

pybgpkitstream/bgpstreamconfig.py

@@ -1,5 +1,5 @@
 import datetime
-from pydantic import BaseModel, Field, DirectoryPath
+from pydantic import BaseModel, Field, DirectoryPath, field_validator
 from typing import Literal
 from ipaddress import IPv4Address, IPv6Address
 
@@ -70,3 +70,13 @@ class BGPStreamConfig(BaseModel):
         default=datetime.timedelta(hours=2),
         description="Interval for the fetch/parse cycle (avoid long prefetch time)",
     )
+
+    @field_validator("start_time", "end_time")
+    @classmethod
+    def normalize_to_utc(cls, dt: datetime.datetime) -> datetime.datetime:
+        # if naive datetime (not timezone-aware) assume it's UTC
+        if dt.tzinfo is None:
+            return dt.replace(tzinfo=datetime.timezone.utc)
+        # if timezone-aware, convert to utc
+        else:
+            return dt.astimezone(datetime.timezone.utc)

{pybgpkitstream-0.1.4.dist-info → pybgpkitstream-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pybgpkitstream
-Version: 0.1.4
+Version: 0.1.6
 Summary: Drop-in replacement for PyBGPStream using BGPKIT
 Author: JustinLoye
 Author-email: JustinLoye <jloye@iij.ad.jp>

pybgpkitstream-0.1.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+pybgpkitstream/__init__.py,sha256=kNfv6bvDkaGKjlw0pr9LWVqOQtIGmIPk-VG1ZCBuA38,163
+pybgpkitstream/bgpelement.py,sha256=7mXSUmWThhIbKy2JVsLchoteve0BshT3uH8cdbAe0Go,1176
+pybgpkitstream/bgpkitstream.py,sha256=VU79Olu1wk_GNBMxsCKCMjMfSU5YoKA9WsCf-4sOMBk,10997
+pybgpkitstream/bgpstreamconfig.py,sha256=tFZEpqOoFMMNbyf6dzKHEtHZnEnm1Gv88hKy6BqTCq8,3299
+pybgpkitstream/cli.py,sha256=E0E1hO0fzGhy1skBopRufdewsiSy_mA-J8Gf2WxBRxo,4214
+pybgpkitstream/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pybgpkitstream-0.1.6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+pybgpkitstream-0.1.6.dist-info/entry_points.txt,sha256=aWhImGlXLtRKkfyJHudcbSp5K5As4ZGL8wXZC0y6q4o,60
+pybgpkitstream-0.1.6.dist-info/METADATA,sha256=yKcQtzIS0JFXppGsnMVnQRlhVp82_u6Es8Q1fgPLFqQ,2953
+pybgpkitstream-0.1.6.dist-info/RECORD,,

pybgpkitstream-0.1.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-pybgpkitstream/__init__.py,sha256=kNfv6bvDkaGKjlw0pr9LWVqOQtIGmIPk-VG1ZCBuA38,163
-pybgpkitstream/bgpelement.py,sha256=7mXSUmWThhIbKy2JVsLchoteve0BshT3uH8cdbAe0Go,1176
-pybgpkitstream/bgpkitstream.py,sha256=n68cE4IPI1Lir0Zjkd8R0yPAjZg3GbbFZEEH7BbUIB4,10708
-pybgpkitstream/bgpstreamconfig.py,sha256=_PHoNhq8lw4QzNKya-KQFQ24dEbTjTkmefFhx0t6K8Q,2873
-pybgpkitstream/cli.py,sha256=E0E1hO0fzGhy1skBopRufdewsiSy_mA-J8Gf2WxBRxo,4214
-pybgpkitstream/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pybgpkitstream-0.1.4.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-pybgpkitstream-0.1.4.dist-info/entry_points.txt,sha256=aWhImGlXLtRKkfyJHudcbSp5K5As4ZGL8wXZC0y6q4o,60
-pybgpkitstream-0.1.4.dist-info/METADATA,sha256=scEKSheKIwDh35g4AGgz5dD_ba1RSKlTEuG-vj5QXBY,2953
-pybgpkitstream-0.1.4.dist-info/RECORD,,