Compare commits
4 Commits
6f3c7bf8d2
..
v0.1.0
| Author | SHA1 | Date | |
|---|---|---|---|
| 6ad0587246 | |||
| 7a19f9acb7 | |||
| 6c502f94f1 | |||
| e102cfa9c8 |
@@ -1,24 +1,5 @@
|
||||
# myice
|
||||
|
||||
## intro
|
||||
|
||||
Avec tout ça on va aller chercher sur MyIce les planning des gamins et générer
|
||||
les pdf qu'on veut
|
||||
|
||||
## install
|
||||
|
||||
with [uv](https://docs.astral.sh/uv/getting-started/installation/):
|
||||
|
||||
```shell
|
||||
uv tool install --extra-index-url https://gitea.parano.ch/api/packages/herel/pypi/simple/ myice
|
||||
```
|
||||
|
||||
with [pipx](https://pipx.pypa.io/stable/installation/):
|
||||
|
||||
```shell
|
||||
pipx install --extra-index-url https://gitea.parano.ch/api/packages/herel/pypi/simple/ myice
|
||||
```
|
||||
|
||||
## récupérer le schedule
|
||||
|
||||
```shell
|
||||
|
||||
+14
-68
@@ -12,8 +12,6 @@ import sys
|
||||
from enum import Enum
|
||||
from pathlib import Path
|
||||
from typing import Annotated
|
||||
from typing import List, Tuple
|
||||
from rl_ai_tools import utils # type: ignore
|
||||
import requests
|
||||
import typer
|
||||
|
||||
@@ -33,11 +31,6 @@ class AgeGroup(str, Enum):
|
||||
u15a = "U15 (A)"
|
||||
|
||||
|
||||
class EventType(str, Enum):
|
||||
game = "game"
|
||||
practice = "practice"
|
||||
|
||||
|
||||
def load_cookies(file: str = "cookies.txt") -> requests.cookies.RequestsCookieJar:
|
||||
cookie_jar_file = Path(file)
|
||||
cj_dict = {}
|
||||
@@ -47,9 +40,8 @@ def load_cookies(file: str = "cookies.txt") -> requests.cookies.RequestsCookieJa
|
||||
return requests.cookies.cookiejar_from_dict(cj_dict)
|
||||
|
||||
|
||||
def save_cookies(file: str = "cookies.txt"):
|
||||
cookie_jar_file = Path(file)
|
||||
with cookie_jar_file.open("w") as f:
|
||||
def save_cookies():
|
||||
with open("cookies.txt", "w") as f:
|
||||
f.write(json.dumps(requests.utils.dict_from_cookiejar(session.cookies)))
|
||||
|
||||
|
||||
@@ -69,12 +61,10 @@ def get_login(local_file: str = "myice.ini") -> tuple[str, str, int]:
|
||||
password = default_config.get("password")
|
||||
userid = default_config.getint("userid")
|
||||
if not username or not password:
|
||||
print(
|
||||
"Error: please configure username/password in ini file", file=sys.stderr
|
||||
)
|
||||
print("Error: please configure username/password in ini file")
|
||||
sys.exit(1)
|
||||
else:
|
||||
print("Error: please configure username/password in ini file", file=sys.stderr)
|
||||
print("Error: please configure username/password in ini file")
|
||||
sys.exit(1)
|
||||
return username, password, userid
|
||||
|
||||
@@ -128,12 +118,12 @@ def wrapper_session(func):
|
||||
session.cookies = load_cookies()
|
||||
session.cookies.clear_expired_cookies()
|
||||
if not session.cookies.get("mih_v3_cookname"):
|
||||
print("login...", file=sys.stderr)
|
||||
print("login...")
|
||||
do_login()
|
||||
save_cookies()
|
||||
_, _, userid = get_login()
|
||||
if not userid:
|
||||
print("get userid...", file=sys.stderr)
|
||||
print("get userid...")
|
||||
userid = get_userid()
|
||||
return func(*args, **kwargs)
|
||||
|
||||
@@ -217,10 +207,11 @@ def schedule(
|
||||
with outfile.open("w") as f:
|
||||
f.write(schedule)
|
||||
else:
|
||||
print("Schedule:", file=sys.stderr)
|
||||
print(schedule)
|
||||
|
||||
|
||||
def os_open(file: str) -> None:
|
||||
def open(file: str) -> None:
|
||||
if os.uname().sysname == "Linux":
|
||||
os.system(f"xdg-open {file}")
|
||||
else:
|
||||
@@ -237,7 +228,7 @@ def get_game_pdf(
|
||||
"""
|
||||
output_filename = f"game_{game_id}.pdf"
|
||||
game_pdf(game_id, Path(output_filename))
|
||||
os_open(output_filename)
|
||||
open(output_filename)
|
||||
|
||||
|
||||
@app.command("practice")
|
||||
@@ -249,16 +240,12 @@ def get_practice_pdf(
|
||||
"""
|
||||
output_filename = f"practice_{game_id}.pdf"
|
||||
practice_pdf(game_id, Path(output_filename))
|
||||
os_open(output_filename)
|
||||
open(output_filename)
|
||||
|
||||
|
||||
@app.command("search")
|
||||
def parse_schedule(
|
||||
age_group: Annotated[AgeGroup | None, typer.Option(...)] = None,
|
||||
event_type_filter: Annotated[
|
||||
EventType | None,
|
||||
typer.Option("--type", help="Only display events of this type"),
|
||||
] = None,
|
||||
age_group: Annotated[AgeGroup, typer.Argument(...)],
|
||||
schedule_file: Annotated[
|
||||
Path, typer.Option(help="schedule json file to parse")
|
||||
] = Path("schedule.json"),
|
||||
@@ -268,22 +255,9 @@ def parse_schedule(
|
||||
"""
|
||||
with schedule_file.open("r") as f:
|
||||
data = json.load(f)
|
||||
# age_group filter
|
||||
if age_group:
|
||||
events = [x for x in data if x["agegroup"] == age_group]
|
||||
else:
|
||||
events = [x for x in data if x["agegroup"] in AgeGroup]
|
||||
# event_type filter
|
||||
if event_type_filter:
|
||||
if event_type_filter.value == EventType.game:
|
||||
events = [x for x in events if "event" in x and x["event"] == "Jeu"]
|
||||
else:
|
||||
events = [x for x in events if "event" not in x or x["event"] == "Jeu"]
|
||||
for event in events:
|
||||
if age_group:
|
||||
raw_title = event["title"].removeprefix(age_group + "\n")
|
||||
else:
|
||||
raw_title = event["title"]
|
||||
for event in [x for x in data if x["agegroup"] == age_group]:
|
||||
# print(json.dumps(event, indent=2))
|
||||
raw_title = event["title"].removeprefix(age_group + "\n")
|
||||
title = " ".join(raw_title.split("\n"))
|
||||
start = datetime.datetime.fromisoformat(event["start"])
|
||||
start_fmt = start.strftime("%H:%M")
|
||||
@@ -299,33 +273,5 @@ def parse_schedule(
|
||||
print(f"{event_type}: {title}\n")
|
||||
|
||||
|
||||
@app.command("ai")
|
||||
def check_with_ai(
|
||||
schedule_file: Annotated[
|
||||
Path, typer.Option(help="schedule json file to parse")
|
||||
] = Path("schedule.json"),
|
||||
):
|
||||
"""
|
||||
Search through the schedule with natural language using Infomaniak LLM API
|
||||
"""
|
||||
if not utils.init():
|
||||
sys.exit(1)
|
||||
with schedule_file.open("r") as f:
|
||||
schedule_data = json.load(f)
|
||||
system = (
|
||||
"Tu es une IA connaissant bien les données suivantes, qui décrivent les match et entraînements d'équipes de hockey sur glace:\n"
|
||||
+ json.dumps(schedule_data)
|
||||
)
|
||||
history: List[Tuple[str, str]] = []
|
||||
while True:
|
||||
try:
|
||||
question = input("> ")
|
||||
except EOFError:
|
||||
break
|
||||
answer = utils.llm_inference(question, history, system=system)
|
||||
print(">", answer)
|
||||
history.append((question, answer))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
app()
|
||||
|
||||
Generated
+1
-21
@@ -890,26 +890,6 @@ pygments = ">=2.13.0,<3.0.0"
|
||||
[package.extras]
|
||||
jupyter = ["ipywidgets (>=7.5.1,<9)"]
|
||||
|
||||
[[package]]
|
||||
name = "rl-ai-tools"
|
||||
version = "1.9.0"
|
||||
description = "Tools using AI"
|
||||
optional = false
|
||||
python-versions = ">=3.11,<4.0"
|
||||
files = [
|
||||
{file = "rl_ai_tools-1.9.0-py3-none-any.whl", hash = "sha256:c986a17e39afe6bd5bc21eb8d259e465dcc52e60a73b81db3931b45599e94fc1"},
|
||||
{file = "rl_ai_tools-1.9.0.tar.gz", hash = "sha256:80c43d434a330544e5d201f3a99460996cc31276fb360cb45a0962f743a342bb"},
|
||||
]
|
||||
|
||||
[package.dependencies]
|
||||
requests = ">=2.32.3,<3.0.0"
|
||||
typer = ">=0.12.5,<0.13.0"
|
||||
|
||||
[package.source]
|
||||
type = "legacy"
|
||||
url = "https://pypi.purple.infomaniak.ch"
|
||||
reference = "infomaniak"
|
||||
|
||||
[[package]]
|
||||
name = "shellingham"
|
||||
version = "1.5.4"
|
||||
@@ -1045,4 +1025,4 @@ files = [
|
||||
[metadata]
|
||||
lock-version = "2.0"
|
||||
python-versions = "^3.12"
|
||||
content-hash = "c8b74100d5c4f5889db217b5fb249f55248d1955787fd76dbda5ded8b73875b3"
|
||||
content-hash = "e11a9960c50f8b210bd0942533a30915eb56bed47cf5f67ac33f405cb1ef5859"
|
||||
|
||||
+2
-9
@@ -1,6 +1,6 @@
|
||||
[tool.poetry]
|
||||
name = "myice"
|
||||
version = "v0.2.0"
|
||||
version = "0.1.0"
|
||||
description = "myice parsing"
|
||||
authors = ["Rene Luria <rene@luria.ch>"]
|
||||
license = "MIT"
|
||||
@@ -9,19 +9,12 @@ readme = "README.md"
|
||||
[tool.poetry.dependencies]
|
||||
python = "^3.12"
|
||||
requests = "^2.32.3"
|
||||
typer = "^0.12.5"
|
||||
rl-ai-tools = {version = "^1.9.0", source = "infomaniak"}
|
||||
typer = {extras = ["all"], version = "^0.12.5"}
|
||||
|
||||
|
||||
[tool.poetry.group.dev.dependencies]
|
||||
ipykernel = "^6.29.5"
|
||||
|
||||
|
||||
[[tool.poetry.source]]
|
||||
name = "infomaniak"
|
||||
url = "https://pypi.purple.infomaniak.ch"
|
||||
priority = "supplemental"
|
||||
|
||||
[build-system]
|
||||
requires = ["poetry-core"]
|
||||
build-backend = "poetry.core.masonry.api"
|
||||
|
||||
Reference in New Issue
Block a user