Skip to content

Commit e351f06

Browse files
authored
Merge pull request #36 from con/user-agent
Set custom User-Agent in HTTP requests
2 parents 9c93332 + 3368838 commit e351f06

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

src/solidation/__main__.py

+13-2
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,11 @@
33
from collections.abc import Iterable, Iterator
44
from dataclasses import InitVar, dataclass, field
55
from datetime import datetime, timedelta, timezone
6+
import importlib.metadata
67
import logging
78
import os
89
from pathlib import Path
10+
import platform
911
from random import sample
1012
import re
1113
from statistics import quantiles
@@ -19,10 +21,19 @@
1921
from github.Repository import Repository
2022
from pydantic import BaseModel, Field, StrictBool, StringConstraints
2123
from ruamel.yaml import YAML
22-
from . import __version__
24+
from . import __url__, __version__
2325

2426
log = logging.getLogger("solidation")
2527

28+
USER_AGENT = "solidation/{} ({}) PyGithub/{} {}/{}".format(
29+
__version__,
30+
__url__,
31+
importlib.metadata.version("pygithub"),
32+
platform.python_implementation(),
33+
platform.python_version(),
34+
)
35+
36+
2637
GHUser = Annotated[str, StringConstraints(pattern=r"^[-_A-Za-z0-9]+$")]
2738
GHRepo = Annotated[str, StringConstraints(pattern=r"^[-_A-Za-z0-9]+/[-_.A-Za-z0-9]+$")]
2839

@@ -76,7 +87,7 @@ class Consolidator:
7687
since: datetime = field(init=False)
7788

7889
def __post_init__(self, token: str) -> None:
79-
self.gh = Github(auth=Auth.Token(token))
90+
self.gh = Github(auth=Auth.Token(token), user_agent=USER_AGENT)
8091
self.since = datetime.now(timezone.utc) - timedelta(
8192
days=self.config.recent_days
8293
)

0 commit comments

Comments
 (0)