Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 25 additions & 26 deletions src/sparsezoo/analytics.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@

import machineid
import requests
from requests import HTTPError

from sparsezoo.utils.gdpr import is_gdpr_country
from sparsezoo.utils.suppress import suppress_stdout_stderr
from sparsezoo.version import version as sparsezoo_version


Expand Down Expand Up @@ -122,31 +122,30 @@ def send_event(
event_params = {}

def _send_request():
with suppress_stdout_stderr(suppress=not _DEBUG):
event_params.update(self._package_params)
event_params["package"] = self._package
event_params["version"] = self._version
payload = {
"client_id": self._client_id,
"events": [{"name": event_name, "params": event_params}],
}
headers = {
"Content-Type": "application/json",
}
data = json.dumps(payload)

try:
response = requests.post(self._url, headers=headers, data=data)
response.raise_for_status()
body = response.content
if _DEBUG:
print(body)
except Exception as err:
if _DEBUG:
print(err)

if raise_errors:
raise err
event_params.update(self._package_params)
event_params["package"] = self._package
event_params["version"] = self._version
payload = {
"client_id": self._client_id,
"events": [{"name": event_name, "params": event_params}],
}
headers = {
"Content-Type": "application/json",
}
data = json.dumps(payload)

try:
response = requests.post(self._url, headers=headers, data=data)
response.raise_for_status()
body = response.content
if _DEBUG:
print(body)
except HTTPError as http_error:
if _DEBUG:
print(http_error)

if raise_errors:
raise http_error

thread = threading.Thread(target=_send_request)
thread.start()
Expand Down
29 changes: 13 additions & 16 deletions src/sparsezoo/utils/gdpr.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,7 @@

import geocoder
import requests

from sparsezoo.utils.suppress import suppress_stdout_stderr
from requests import HTTPError


__all__ = ["get_external_ip", "get_country_code", "is_gdpr_country"]
Expand Down Expand Up @@ -58,28 +57,26 @@ def get_external_ip() -> Optional[str]:
"""
:return: the external ip of the machine, None if unable to get
"""
with suppress_stdout_stderr():
try:
response = requests.get("https://ident.me")
external_ip = response.text.strip()
try:
response = requests.get("https://ident.me")
external_ip = response.text.strip()

return external_ip
except Exception:
return None
return external_ip
except Exception:
return None


def get_country_code() -> Optional[str]:
"""
:return: the country code of the machine, None if unable to get
"""
with suppress_stdout_stderr():
try:
ip = get_external_ip()
geo = geocoder.ip(ip)
try:
ip = get_external_ip()
geo = geocoder.ip(ip)

return geo.country
except Exception:
return None
return geo.country
except HTTPError:
return None


def is_gdpr_country() -> bool:
Expand Down
45 changes: 0 additions & 45 deletions src/sparsezoo/utils/suppress.py

This file was deleted.