Skip to content

Commit facd2f2

Browse files
remove pkg_resources
1 parent ed591da commit facd2f2

File tree

4 files changed

+7
-8
lines changed

4 files changed

+7
-8
lines changed

geoengine/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
'''Entry point for Geo Engine Python Library'''
22

3-
from pkg_resources import get_distribution
43
from requests import utils
54
from pydantic import ValidationError
65
from geoengine_openapi_client.exceptions import BadRequestException, OpenApiException, ApiTypeError, ApiValueError, \
76
ApiKeyError, ApiAttributeError, ApiException, NotFoundException
87
from geoengine_openapi_client import UsageSummaryGranularity
8+
import geoengine_openapi_client
99

1010
from . import workflow_builder
1111
from .raster_workflow_rio_writer import RasterWorkflowRioWriter
@@ -39,7 +39,7 @@
3939
add_or_replace_dataset_with_permissions, dataset_info_by_name
4040

4141

42-
DEFAULT_USER_AGENT = f'geoengine-python/{get_distribution("geoengine").version}'
42+
DEFAULT_USER_AGENT = f'geoengine-python/{geoengine_openapi_client.__version__}'
4343

4444

4545
def default_user_agent(_name="python-requests"):

geoengine/auth.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
import os
1010
from dotenv import load_dotenv
11-
from pkg_resources import get_distribution
1211
from requests.auth import AuthBase
1312
import urllib3
1413

@@ -71,7 +70,7 @@ def __init__(self,
7170
# Auto-generated SessionApi cannot handle dynamically differing return types (SimpleSession or UserSession).
7271
# Because of that requests must be send manually.
7372
http = urllib3.PoolManager()
74-
user_agent = f'geoengine-python/{get_distribution("geoengine").version}'
73+
user_agent = f'geoengine-python/{geoengine_openapi_client.__version__}'
7574

7675
if credentials is not None:
7776
session = http.request(

tests/test_auth.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from datetime import datetime
44
import unittest
55
import os
6-
from pkg_resources import get_distribution
6+
import geoengine_openapi_client
77
import geoengine as ge
88
from geoengine.error import GeoEngineException
99
from geoengine.types import QueryRectangle
@@ -120,7 +120,7 @@ def test_initialize_env(self):
120120
def test_user_agent(self):
121121
with UrllibMocker() as m:
122122
m.post('http://mock-instance/anonymous',
123-
request_headers={'User-Agent': f'geoengine-python/{get_distribution("geoengine").version}'},
123+
request_headers={'User-Agent': f'geoengine-python/{geoengine_openapi_client.__version__}'},
124124
json={
125125
"id": "e327d9c3-a4f3-4bd7-a5e1-30b26cae8064",
126126
"user": None,

tests/test_wfs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import geopandas as gpd
99
import geopandas.testing # pylint: disable=unused-import
1010
from shapely.geometry import Point
11-
from pkg_resources import get_distribution
11+
import geoengine_openapi_client
1212
import geoengine as ge
1313
from . import UrllibMocker
1414

@@ -829,7 +829,7 @@ def test_owslib_user_agent(self):
829829
}]
830830
},
831831
request_headers={'Authorization': 'Bearer e327d9c3-a4f3-4bd7-a5e1-30b26cae8064',
832-
'User-Agent': f'geoengine/openapi-client/python/{get_distribution("geoengine-openapi-client").version}'}
832+
'User-Agent': f'geoengine/openapi-client/python/{geoengine_openapi_client.__version__}'}
833833
)
834834

835835
ge.initialize("http://mock-instance")

0 commit comments

Comments
 (0)