Skip to content

Commit

Permalink
pre-commit file changes
Browse files Browse the repository at this point in the history
  • Loading branch information
hmkamel committed May 9, 2024
1 parent 3d17a96 commit f68b66f
Show file tree
Hide file tree
Showing 9 changed files with 74 additions and 27 deletions.
2 changes: 1 addition & 1 deletion .flake8
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[flake8]
ignore = E203, E266, E501, W503, F403, F401
ignore = E203, E266, E501, W503, F403, F401, F821
max-line-length = 79
max-complexity = 18
select = B,C,E,F,W,T4,B9
2 changes: 1 addition & 1 deletion .idea/misc.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -13,4 +13,4 @@ exclude = '''
| build
| dist
)/
'''
'''
34 changes: 23 additions & 11 deletions src/iracing_connect/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,9 @@ def __init__(self, transport, logger):
self.logger = logger
self.transport = transport

def _get(self, url: str, api_group: str, func_name: str, params=None) -> dict:
def _get(
self, url: str, api_group: str, func_name: str, params=None
) -> dict:
response = self.transport.get(url=url, params=params)

if response.status_code == requests.codes.OK:
Expand Down Expand Up @@ -80,8 +82,6 @@ def _get(self, url: str, params: dict, has_chunks: bool = False):

payload_dict = json.loads(response_text)

payload = json.dumps(payload_dict) # TODO: delete

print()

return payload_dict
Expand All @@ -102,9 +102,9 @@ def _get_chunks(self, payload):
allow_redirects=False,
timeout=10.0,
).text
except:
except: # noqa
# TODO: build exceptions
msg = f"API extraction error at {chunk_file_name}, {i}/{len(chunk_file_names)}"
msg = f"API extraction error at {chunk_file_name}, {i}/{len(chunk_file_names)}" # noqa
pass

chunks.append(response)
Expand All @@ -130,7 +130,9 @@ def member_bests(self, customer_id, car_id):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record

def member_career(self, customer_id):
Expand All @@ -139,7 +141,9 @@ def member_career(self, customer_id):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record

def member_recent_races(self, customer_id):
Expand All @@ -148,7 +152,9 @@ def member_recent_races(self, customer_id):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record

def member_summary(self, customer_id):
Expand All @@ -157,7 +163,9 @@ def member_summary(self, customer_id):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record

def member_yearly(self, customer_id):
Expand All @@ -166,7 +174,9 @@ def member_yearly(self, customer_id):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record

def membership(self, customer_id: int):
Expand All @@ -175,5 +185,7 @@ def membership(self, customer_id: int):
func_name = inspect.stack()[0][3]

results_response = self._get(endpoint, params)
record = self._wrap_payload(results_response, func_name, endpoint, params)
record = self._wrap_payload(
results_response, func_name, endpoint, params
)
return record
4 changes: 3 additions & 1 deletion src/iracing_connect/endpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,9 @@
URL_STATS_MEMBER_CAREER = URL + "/data/stats/member_career"
URL_STATS_MEMBER_SUMMARY = URL + "/data/stats/member_summary"
URL_STATS_MEMBER_YEARLY = URL + "/data/stats/member_yearly"
URL_STATS_MEMBER_RECENT_RACES = URL + "/data/stats/member_recent_races" # ingest
URL_STATS_MEMBER_RECENT_RACES = (
URL + "/data/stats/member_recent_races"
) # ingest


# league
Expand Down
4 changes: 3 additions & 1 deletion src/iracing_connect/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,9 @@ def get_current_utc_time():


def encode_pw(username, password):
initialHash = hashlib.sha256((password + username.lower()).encode("utf-8")).digest()
initialHash = hashlib.sha256(
(password + username.lower()).encode("utf-8")
).digest()
hashInBase64 = base64.b64encode(initialHash).decode("utf-8")

return hashInBase64
24 changes: 19 additions & 5 deletions src/iracing_connect/results/results_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,15 @@ def event_log(self, subsession_id: int, simsession_number: int) -> dict:
}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def lap_chart_data(self, subsession_id: int, simsession_number: int) -> dict:
def lap_chart_data(
self, subsession_id: int, simsession_number: int
) -> dict:
endpoint = endpoints.URL_RESULTS_LAP_CHART_DATA
func_name = inspect.stack()[0][3]

Expand All @@ -32,7 +37,10 @@ def lap_chart_data(self, subsession_id: int, simsession_number: int) -> dict:
}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def lap_data(self, subsession_id: int, simsession_number: int) -> dict:
Expand All @@ -45,7 +53,10 @@ def lap_data(self, subsession_id: int, simsession_number: int) -> dict:
}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def summary(self, subsession_id: int, simsession_number: int) -> dict:
Expand All @@ -58,5 +69,8 @@ def summary(self, subsession_id: int, simsession_number: int) -> dict:
}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)
25 changes: 20 additions & 5 deletions src/iracing_connect/stats/stats_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@ def member_bests(self, customer_id: int, car_id: int) -> dict:
params = {"customer_id": customer_id, "car_id": car_id}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def member_career(self, customer_id: int) -> dict:
Expand All @@ -26,7 +29,10 @@ def member_career(self, customer_id: int) -> dict:
params = {"customer_id": customer_id}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def member_summary(self, customer_id: int) -> dict:
Expand All @@ -36,7 +42,10 @@ def member_summary(self, customer_id: int) -> dict:
params = {"customer_id": customer_id}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def member_yearly(self, customer_id: int) -> dict:
Expand All @@ -46,7 +55,10 @@ def member_yearly(self, customer_id: int) -> dict:
params = {"customer_id": customer_id}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)

def member_recent_races(self, customer_id: int) -> dict:
Expand All @@ -56,5 +68,8 @@ def member_recent_races(self, customer_id: int) -> dict:
params = {"customer_id": customer_id}

return self._get(
url=endpoint, api_group=self.api_group, func_name=func_name, params=params
url=endpoint,
api_group=self.api_group,
func_name=func_name,
params=params,
)
4 changes: 3 additions & 1 deletion src/iracing_connect/transport.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,9 @@ def _get_cookies(self):
def get(self, url: str, params: dict = None):
response = self.request(url, HTTPMethod.GET, params)
self.logger.info(f"Request sent for URL: {response.url}")
self.logger.info(f"Status code of the response: {response.status_code}")
self.logger.info(
f"Status code of the response: {response.status_code}"
)
self.logger.debug(f"Response object: {response.__dict__}")

return response
Expand Down

0 comments on commit f68b66f

Please sign in to comment.