keihan-tracker 2.2.0__py3-none-any.whl

keihan_tracker/__init__.py

@@ -0,0 +1,5 @@
+from .keihan_train import KHTracker
+from .keihan_train.tracker import TrainData, ActiveTrainData, LineLiteral
+from .keihan_train.schemes import TrainType
+from .bus import get_khbus_info
+from .delay_tracker import get_yahoo_delay

keihan_tracker/bus/__init__.py

@@ -0,0 +1 @@
+from .tracker import get_khbus_info

keihan_tracker/bus/schemes.py

@@ -0,0 +1,58 @@
+from typing import List, Optional
+from pydantic import BaseModel, model_validator
+
+
+class Head(BaseModel):
+    errorcode: int
+    objid: str
+
+class BusStatePrms(BaseModel):
+    order: int                 # 接近順位
+    lat: float                 # 緯度
+    lon: float                 # 経度
+    heading: int               # 方位
+    guid: str                  # GUID
+    stop_id: int               # 停留所ID？
+    platform_index: int        # のりばインデックス？
+    route: str                 # 系統番号（例: [22]）
+    destination: str           # 行先
+    via: str                   # 経由
+    status: str                # 状態（到着済など）
+    timetable: str             # 時刻文字列（例: 17:30 到着予定）
+
+    @classmethod
+    def from_string(cls, raw: str):
+        parts = raw.split(":")
+        return cls(
+            order=int(parts[0]),
+            lat=float(parts[1]),
+            lon=float(parts[2]),
+            heading=int(parts[3]),
+            guid=parts[4],
+            stop_id=int(parts[5]),
+            platform_index=int(parts[6]),
+            route=parts[7],
+            destination=parts[8],
+            via=parts[9],
+            status=parts[10],
+            timetable=parts[11].replace("__________", ":"),
+        )
+
+class BusState(BaseModel):
+    html: Optional[str] = None
+    html_sp: Optional[str] = None
+    busstateprms: BusStatePrms
+
+    @model_validator(mode="before")
+    def parse_prms(cls, values):
+        if isinstance(values, dict) and "busstateprms" in values and isinstance(values["busstateprms"], str):
+            values["busstateprms"] = BusStatePrms.from_string(values["busstateprms"])
+        return values
+
+class Body(BaseModel):
+    datetimeStr: str
+    busstates: List[BusState]
+
+class BusLocationResponse(BaseModel):
+    head: Head
+    body: Body

keihan_tracker/bus/tracker.py

@@ -0,0 +1,13 @@
+from httpx import AsyncClient
+import urllib.parse
+from keihan_tracker.bus.schemes import BusLocationResponse
+
+UPDATE_URL = "https://busnavi.keihanbus.jp/pc/busstateupd"
+
+async def get_khbus_info(stop_name:str, stop_num:int=1):
+    async with AsyncClient() as client:
+        dgmpl = f"{stop_name}:{stop_num}::"
+        result = await client.post(UPDATE_URL,data={"dgmpl":urllib.parse.quote(dgmpl), "sort4":"0"})
+        result.raise_for_status()
+        res = BusLocationResponse.model_validate_json(result.text)
+        return res

keihan_tracker/delay_tracker.py

@@ -0,0 +1,153 @@
+from pydantic import BaseModel, field_validator, BeforeValidator, Field
+from typing import Optional, Union, Annotated, List, Any
+from datetime import datetime
+from zoneinfo import ZoneInfo
+from bs4 import BeautifulSoup as bs
+import asyncio
+import httpx
+import re
+from urllib.parse import urljoin
+
+JST = ZoneInfo("Asia/Tokyo")
+BASE_URL = "https://transit.yahoo.co.jp/diainfo/area/"
+
+class DelayLine(BaseModel):
+    LineName: str
+    status: str
+    detail: str
+    AnnouncedTime: datetime
+
+    def __str__(self):
+        return f"{self.LineName}: {self.status}\n{self.detail}\n{self.AnnouncedTime} 発表"
+
+def force_list(v: Any) -> List[Any]:
+    if v is None: return []
+    if isinstance(v, list): return v
+    return [v]
+
+AsList = BeforeValidator(force_list)
+
+# --- モデル定義 (クラス名を変更して衝突を回避) ---
+
+class LineData(BaseModel):
+    Name: str = ""
+    code: str = ""
+    corporationIndex: str = ""
+
+class CommentData(BaseModel):
+    text: str = ""
+    status: str = ""
+
+class PrefectureData(BaseModel):
+    Name: str = ""
+    code: str = ""
+
+class StationData(BaseModel):
+    Name: str = ""
+    code: str = ""
+    Type: str = ""
+    Yomi: str = ""
+
+class PointData(BaseModel):
+    # フィールド名はJSONに合わせて "Prefecture" だが、型は "PrefectureData" にする
+    Prefecture: PrefectureData = Field(default_factory=PrefectureData)
+    Station: StationData = Field(default_factory=StationData)
+
+class SectionData(BaseModel):
+    Point: Annotated[List[PointData], AsList] = []
+
+class CorporationData(BaseModel):
+    Name: str = ""
+    code: str = ""
+
+class InformationData(BaseModel):
+    status: str = ""
+    provider: str = ""
+    Title: str = ""
+    Datetime: datetime
+    
+    # ここでもフィールド名と型名を区別する
+    Line: LineData = Field(default_factory=LineData)
+    
+    Section: Annotated[List[SectionData], AsList] = []
+    Comment: Annotated[List[CommentData], AsList] = []
+    Prefecture: Annotated[List[PrefectureData], AsList] = []
+
+class ResultSetData(BaseModel):
+    apiVersion: str = ""
+    engineVersion: str = ""
+    Information: Annotated[List[InformationData], AsList] = []
+    Corporation: Annotated[List[CorporationData], AsList] = []
+
+class ResponseModel(BaseModel):
+    ResultSet: ResultSetData
+
+
+async def get_yahoo_delay(area:int=6) -> list[DelayLine]:
+    async with httpx.AsyncClient() as crowler:
+        html = await crowler.get(f"{BASE_URL}{area}")
+
+        soup = bs(html.content, "html.parser")
+        
+        table = soup.select("div.elmTblLstLine.trouble")[0].find_all("table")
+        table = table[0] if len(table) != 0 else None
+        if not table:
+            return []
+        
+        async def get_detail(url:str) -> tuple[str,str]:
+            res = await crowler.get(url,follow_redirects=True)
+            soup = bs(res.text, "html.parser")
+            info = soup.select("div#mdServiceStatus")[0]
+            title = info.select("dl > dt")[0].text
+            text = info.select("dl > dd > p")[0].text
+
+            return title, text
+
+        delays:list[DelayLine] = []
+        tasks:list = []
+        lines:list[tuple[str,str]] = []
+        for tr in table.find_all("tr")[1:]:
+            url = urljoin(BASE_URL, str(tr.find_all("td")[0].find_all("a")[0].get("href")))
+            line = tr.find_all("td")[0].text    # ○○線
+            short_status = tr.find_all("td")[2].text  # 17:00頃、宇都宮...
+            tasks.append(get_detail(url))
+            lines.append((line, short_status))
+
+        for i, line in zip(await asyncio.gather(*tasks), lines):
+            line, status = line
+            title, text = i
+            
+            pattern = r"（(?P<month>\d{1,2})月(?P<day>\d{1,2})日\s*(?P<hour>\d{1,2})時(?P<minute>\d{1,2})分掲載）"
+            m = re.search(pattern, text) or {}
+
+            dt = datetime(
+                year=datetime.now().year,  # 年は別途補完
+                month=int(m["month"]),
+                day=int(m["day"]),
+                hour=int(m["hour"]),
+                minute=int(m["minute"]),
+                tzinfo=JST
+            )
+            delays.append(DelayLine(LineName=line, status=title, detail=text, AnnouncedTime=dt))
+        return delays
+    
+async def get_ekispert_delay(api_key:str, prefs:list[int]=[26,27,28]) -> list[DelayLine]:
+    async with httpx.AsyncClient() as web:
+        uri = f"http://api.ekispert.jp/v1/json/operationLine/service/rescuenow/information?key={api_key}"    
+        uri += f"&prefectureCode={':'.join(map(str,prefs))}"
+        request = await web.get(uri)
+        request.raise_for_status()
+        res = ResponseModel.model_validate(request.json())
+        
+        results:dict[str,DelayLine] = {}
+        for i in res.ResultSet.Information or []:
+            if i.Line.code in results:
+                if i.Datetime < (results[i.Line.code].AnnouncedTime or datetime.min):
+                    continue
+            results[i.Line.code] = DelayLine(
+                LineName=i.Line.Name,
+                status=i.status,
+                detail=i.Comment[0].text,
+                AnnouncedTime=i.Datetime
+            )
+        return list(results.values())

keihan_tracker/keihan_train/__init__.py

@@ -0,0 +1 @@
+from .tracker import KHTracker

keihan_tracker/keihan_train/position_calculation.py

@@ -0,0 +1,129 @@
+# GEMINI VIBE CODING
+
+from typing import Optional
+from .schemes import LineLiteral
+
+def calc_position(col: int, row: int) -> tuple[LineLiteral, int, Optional[int]]:
+    """座標から、停車中の駅番号、もしくは2駅の駅番号を返します。"""
+    line: LineLiteral
+    
+    # --- 1. 路線判定 ---
+    if 1 <= row <= 131:
+        # Row 118以降かつcol 1,2は中之島線
+        if row >= 118 and (col == 1 or col == 2):
+            line = "中之島線"
+        else:
+            line = "京阪本線・鴨東線"
+    elif 132 <= row <= 153:
+        line = "宇治線"
+    elif 154 <= row <= 175:
+        line = "交野線"
+    else:
+        raise ValueError(f"Inputed row {row} is out of range.")
+    
+    # colのバリデーション
+    if col <= 0 or col >= 6:
+        raise ValueError(f"Inputed col {col} is out of range.")
+    
+
+    # --- 2. 駅IDと状態の計算 ---
+
+    # === A. 京阪本線・鴨東線・中之島線 (Row 1-131) ===
+    if row <= 131:
+        # A-1. 通常区間 (Row 1-117)
+        if row <= 117:
+            block_index = (row - 1) // 3
+            current_station = 42 - block_index
+            pos_in_block = (row - 1) % 3  # 0:Top, 1:Mid, 2:Bot
+            
+            is_stopped = False
+            
+            # 基本: Top(0)は停車
+            if pos_in_block == 0:
+                is_stopped = True
+            
+            # 特例: 2行目(Mid)以降も停車扱いになる駅 (HTMLの big-area/special-area 等に基づく)
+            # 出町柳(42): Row 2も停車
+            if current_station == 42 and pos_in_block == 1: is_stopped = True
+            # 枚方市(21): Row 65(Mid)も停車
+            if current_station == 21 and pos_in_block == 1: is_stopped = True
+            # 京橋(4): Row 116(Mid)も停車
+            if current_station == 4 and pos_in_block == 1: is_stopped = True
+            
+            if is_stopped:
+                return (line, current_station, None)
+            else:
+                # 移動中 (当駅 -> 次駅)
+                # Next station is current - 1
+                return (line, current_station, current_station - 1)
+
+        # A-2. 地下・中之島線区間 (Row 118-131)
+        # この区間はHTML上隙間なく special-area が続くため、すべて停車扱いとする
+        else:
+            # 共通: 天満橋(3) Row 118-120
+            if row <= 120: return (line, 3, None)
+
+            if line == "中之島線":
+                if row <= 123: return (line, 51, None) # なにわ橋
+                elif row <= 126: return (line, 52, None) # 大江橋
+                elif row <= 129: return (line, 53, None) # 渡辺橋
+                else: return (line, 54, None) # 中之島
+            
+            else: # 京阪本線 大阪側
+                if row <= 123: return (line, 2, None) # 北浜
+                elif row <= 126: return (line, 1, None) # 淀屋橋
+                # Row 127以降は本線には無し
+                else: raise ValueError(f"Invalid row {row} for Main Line")
+
+    # === B. 宇治線 (Row 132-153) ===
+    elif row <= 153:
+        # 中書島(28) Row 132-134 (始発)
+        if row <= 134:
+            # HTML: 132(special), 133(big) -> 停車
+            pos = row - 132
+            if pos <= 1: return (line, 28, None)
+            else: return (line, 28, 71) # 発車
+            
+        # 終点: 宇治(77)
+        if row == 153: return (line, 77, None)
+            
+        # 通常駅 (KH71-KH76)
+        block_idx = (row - 135) // 3
+        current_station = 71 + block_idx
+        pos_in_block = (row - 135) % 3
+        
+        # 基本: Top(0)のみ停車
+        if pos_in_block == 0:
+            return (line, current_station, None)
+        else:
+            return (line, current_station, current_station + 1)
+
+    # === C. 交野線 (Row 154-175) ===
+    elif row <= 175:
+        # 枚方市(21) Row 154-156 (始発)
+        if row <= 156:
+            # HTML: 154(special), 155(big) -> 停車
+            pos = row - 154
+            if pos <= 1: return (line, 21, None)
+            else: return (line, 21, 61) # 発車
+            
+        # 終点: 私市(67)
+        if row == 175: return (line, 67, None)
+            
+        # 通常駅 (KH61-KH66)
+        block_idx = (row - 157) // 3
+        current_station = 61 + block_idx
+        pos_in_block = (row - 157) % 3
+        
+        # 基本: Top(0)のみ停車
+        if pos_in_block == 0:
+            return (line, current_station, None)
+        else:
+            return (line, current_station, current_station + 1)
+
+    return (line, 0, None)
+
+if __name__ == "__main__":
+    col,row=map(int,input().split())
+    r = calc_position(col,row)
+    print(r)

keihan_tracker/keihan_train/schemes.py

@@ -0,0 +1,214 @@
+"""
+京阪電車リアルタイム列車位置情報APIのデータを、Pythonで安全かつ型安全に扱うためのPydanticモデル群。
+
+【特徴】
+・APIから取得したJSONをバリデート（型検証）することが主目的。
+・API仕様に忠実で、余計な変換やロジックを加えず「生に近い」データ構造を維持。
+※臨時列車判定は例外的に実装。
+
+【バリデートできるJSON APIと内容】
+------------------------------------------------------------
+
+1. 駅ごとの乗り換え情報
+   URL: https://www.keihan.co.jp/zaisen/transferGuideInfo.json
+   内容: 各駅で接続している他路線（京阪・地下鉄・モノレール等）の情報。駅番号をキーに、接続路線名（多言語対応）などが格納されている。
+   対応クラス: TransferGuideInfo > StationConnections > MultiLang_Lines
+
+
+2. 路線ごとの駅名データ
+   URL: https://www.keihan.co.jp/zaisen/select_station.json
+   内容: 路線ごとに、駅番号・駅名（多言語対応）などをまとめたデータ。路線名をキーに、各駅の情報が格納されている。
+   対応クラス: SelectStation > LineDetail > MultiLang
+
+
+3. 列車のリアルタイム走行位置
+   URL: https://www.keihan.co.jp/zaisen-up/trainPositionList.json
+   内容: 現在運行中の列車の位置・種別・遅延情報など。各列車の現在地や行先、種別、遅延分数などが含まれる。
+   対応クラス: trainPositionList > LocationObject > trainInfoObject
+
+
+4. 列車ごとのダイヤ（停車駅・時刻表）
+   URL: https://www.keihan.co.jp/zaisen-up/startTimeList.json
+   内容: 各列車の停車駅・発車時刻などのダイヤ情報。列車ごとに、どの駅に何時何分に停車するか等が記載されている。
+   対応クラス: startTimeList > TrainInfo > diaStationInfoObject
+"""
+
+from pydantic import BaseModel, RootModel, Field, model_validator, field_validator
+from typing import Optional, Literal
+from enum import Enum
+import datetime
+from zoneinfo import ZoneInfo
+
+JST = ZoneInfo("Asia/Tokyo")
+
+# 列車種別をEnumで定義
+class TrainType(str, Enum):
+    LOCAL = "普通"
+    SEMI_EXP = "区間急行"
+    SUB_EXP = "準急"
+    COMMUTER_SUB_EXP = "通勤準急"
+    EXPRESS = "急行"
+    COMMUTER_EXP = "通勤急行"
+    MIDNIGHT_EXP = "深夜急行"
+    RAPID_EXP = "快速急行"
+    COMMUTER_RAPID_EXP = "通勤快急"
+    LTD_EXP = "特急"
+    LINER = "ライナー"
+    RAPID_LTD_EXP = "快速特急 洛楽"
+    EXTRA_TRAIN = "臨時列車"  # 臨時列車は臨時特急や臨時休校などとは別
+
+LineLiteral = Literal["京阪本線・鴨東線","中之島線","交野線","宇治線"]
+    
+# 1. 言語別の路線名リストを表現するモデル
+class MultiLang_Lines(BaseModel):
+    """
+    各言語での路線名を保持するモデル。
+    """
+    ja: list[str] = Field(description="日本語での名称")
+    en: list[str] = Field(description="英語での名称")
+    cn: list[str] = Field(description="中国語（簡体字）での名称")
+    tw: list[str] = Field(description="中国語（繁体字）での名称")
+    kr: list[str] = Field(description="韓国語での名称")
+
+# 1. 言語名称モデル
+class MultiLang(BaseModel):
+    """
+    各言語での名称を保持するモデル。
+    """
+    ja: str = Field(description="日本語での名称")
+    en: str = Field(description="英語での名称")
+    cn: str = Field(description="中国語（簡体字）での名称")
+    tw: str = Field(description="中国語（繁体字）での名称")
+    kr: str = Field(description="韓国語での名称")
+
+# 2. 交通手段ごとの接続情報を表現するモデル
+class StationConnections(BaseModel):
+    """
+    駅で接続する交通手段（電車、地下鉄、モノレール）とその路線名を保持するモデル。
+    各交通手段は存在しない場合があるため、Optionalとして定義します。
+    """
+    train:      Optional[MultiLang_Lines] = None
+    subway:     Optional[MultiLang_Lines] = None
+    monorail:   Optional[MultiLang_Lines] = None
+
+### 3. JSONのパーサー
+class TransferGuideInfo(RootModel[dict[str, StationConnections]]):
+    pass
+
+# 1. 駅名データ 
+class LineDetail(BaseModel):
+    lineName: MultiLang
+    stations: dict[str, MultiLang]
+
+### 2. JSONのパーサー
+class SelectStation(
+    RootModel[
+        dict[
+            LineLiteral,
+            LineDetail
+        ]
+            ]):
+    pass
+
+# 1. trainInfoObjectsの要素を表現するモデル
+
+class trainInfoObject(BaseModel):
+    """
+    個々の電車の詳細情報を保持するモデル。
+    """
+    wdfBlockNo:         int
+    carsOfTrain:        int
+    delayMinutes:       str
+    delayMinutesEn:     str
+    delayMinutesKo:     str
+    delayMinutesZhCn:   str
+    delayMinutesZhTw:   str
+    destStationCode:    int
+    destStationNameEn:  str
+    destStationNameJp:  str
+    destStationNameKo:  str
+    destStationNameZhCn:str
+    destStationNameZhTw:str
+    destStationNumber:  int
+    lastPassStation:    int
+    trainNumber:        str
+    trainTypeEn:        str
+    trainTypeIcon:      str
+    trainTypeJp:        TrainType
+    is_special:         bool
+    trainTypeKo:        str
+    trainTypeZhCn:      str
+    trainTypeZhTw:      str
+
+    @model_validator(mode="before")
+    @classmethod
+    def check_type_special(cls, d:dict):
+        if "臨時" in d["trainTypeJp"]:
+            d["is_special"] = True
+            d["trainTypeJp"] = d["trainTypeJp"].replace("臨時　　　　", "")
+        else:
+            d["is_special"] = False
+        return d
+
+# 2. locationObjectsの要素を表現するモデル
+class LocationObject(BaseModel):
+    """
+    路線図上での電車の位置と基本情報を保持するモデル。
+    """
+    delay: str
+    delayEn: str
+    delayKo: str
+    delayZhCn: str
+    delayZhTw: str
+    locationCol: int
+    locationRow: int
+    trainDirection: int
+    trainIconTypeImageJp: str
+    trainInfoObjects: list[trainInfoObject]
+    trainTypeVisIconVis: str
+
+### 3. JSONのパーサー
+class trainPositionList(BaseModel):
+    fileCreatedTime: datetime.datetime
+    fileVersion: str
+    linkNum: str
+    locationObjects: list[LocationObject]
+
+    @field_validator("fileCreatedTime", mode="before")
+    def validate_time(cls,value) -> datetime.datetime:
+        return datetime.datetime.strptime(value,"%Y%m%d%H%M%S").replace(tzinfo=JST)
+
+# 1.
+class diaStationInfoObject(BaseModel):
+    stationNumber:  str = Field(description="駅ナンバリング2桁+ホーム番号1桁 ")
+    stationDepTime: str = Field(description="00:00の形式。深夜は25時などで表す。出発駅の場合は-（ハイフン）、不明な場合は99:99。")
+    stationNameJp:  str
+    stationNameEn:  str
+    stationNameZhTw:str
+    stationNameZhCn:str
+    stationNameKo:  str
+
+class TrainInfo(BaseModel):
+    wdfBlockNo:int
+    extTrain:bool
+    premiumCar:int = Field(description="プレミアムカーがあるかどうか",)
+    trainCar:str = Field(description="車両番号")
+    diaStationInfoObjects:list[diaStationInfoObject]
+
+### 3. JSONのパーサー
+class startTimeList(BaseModel):
+    fileCreatedTime:  datetime.datetime
+    fileVersion:      str
+    TrainInfo:        list[TrainInfo]
+
+    @field_validator("fileCreatedTime", mode="before")
+    def validate_time(cls,value) -> datetime.datetime:
+        return datetime.datetime.strptime(value,"%Y%m%d%H%M%S").replace(tzinfo=JST)
+
+# FileList.xmlのモデル
+class FileList(BaseModel):
+    time: datetime.datetime
+    traininfo: str
+    image_PC: str
+    image_SP: str
+    html_FP: str

keihan_tracker/keihan_train/stations_map.py

@@ -0,0 +1,8 @@
+# ./tracker.py next_stop_station関数の探索用
+# 路線の方面別の駅リスト
+KYOBASHI_TO_DEMACHIYANAGI=[i for i in range(4,42+1)]
+
+UJI_UP = [77,76,75,74,73,72,71,28]
+KATANO_UP = [67,66,65,64,63,62,61,21]
+HONNSEN_UP = [1,2,3]             +KYOBASHI_TO_DEMACHIYANAGI
+NAKANOSHIMA_UP = [54,53,52,51,3] +KYOBASHI_TO_DEMACHIYANAGI