garf-core 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

garf_core/__init__.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.0.6'
+__version__ = '0.0.8'

garf_core/api_clients.py

@@ -19,6 +19,7 @@ import abc
 import dataclasses
 from collections.abc import Sequence
 
+import requests
 from typing_extensions import override
 
 
@@ -34,16 +35,40 @@ class GarfApiResponse:
   results: list
 
 
+class GarfApiError(Exception):
+  """API specific exception."""
+
+
 class BaseClient(abc.ABC):
   """Base API client class."""
 
   @abc.abstractmethod
   def get_response(
-    self, request: GarfApiRequest = GarfApiRequest()
+    self, request: GarfApiRequest = GarfApiRequest(), **kwargs: str
   ) -> GarfApiResponse:
     """Method for getting response."""
 
 
+class RestApiClient(BaseClient):
+  """Specifies REST client."""
+
+  OK = 200
+
+  def __init__(self, endpoint: str, **kwargs: str) -> None:
+    """Initializes RestApiClient."""
+    self.endpoint = endpoint
+    self.query_args = kwargs
+
+  @override
+  def get_response(
+    self, request: GarfApiRequest = GarfApiRequest(), **kwargs: str
+  ) -> GarfApiResponse:
+    response = requests.get(f'{self.endpoint}/{request.resource_name}')
+    if response.status_code == self.OK:
+      return GarfApiResponse(response.json())
+    raise GarfApiError('Failed to get data from API')
+
+
 class FakeApiClient(BaseClient):
   """Fake class for specifying API client."""
 

garf_core/report_fetcher.py

@@ -38,21 +38,30 @@ class ApiReportFetcher:
 
   Attributes:
       api_client: a client used for connecting to API.
+      parser: Type of parser to convert API response.
+      query_specification_builder: Class to perform query parsing.
   """
 
   def __init__(
     self,
     api_client: api_clients.BaseApiClient,
     parser: parsers.BaseParser = parsers.ListParser,
+    query_specification_builder: query_editor.QuerySpecification = (
+      query_editor.QuerySpecification
+    ),
+    **kwargs: str,
   ) -> None:
     """Instantiates ApiReportFetcher based on provided api client.
 
     Args:
       api_client: Instantiated api client.
-      parser: Parser.
+      parser: Type of parser to convert API response.
+      query_specification_builder: Class to perform query parsing.
     """
     self.api_client = api_client
     self.parser = parser()
+    self.query_specification_builder = query_specification_builder
+    self.query_args = kwargs
 
   async def afetch(
     self,
@@ -63,12 +72,12 @@ class ApiReportFetcher:
     """Asynchronously fetches data from API based on query_specification.
 
     Args:
-        query_specification: Query text that will be passed to API
-            alongside column_names, customizers and virtual columns.
-        args: Arguments that need to be passed to the query.
+      query_specification: Query text that will be passed to API
+        alongside column_names, customizers and virtual columns.
+      args: Arguments that need to be passed to the query.
 
     Returns:
-        GarfReport with results of query execution.
+      GarfReport with results of query execution.
     """
     return self.fetch(query_specification, args, **kwargs)
 
@@ -81,19 +90,19 @@ class ApiReportFetcher:
     """Fetches data from API based on query_specification.
 
     Args:
-        query_specification: Query text that will be passed to API
-            alongside column_names, customizers and virtual columns.
-        args: Arguments that need to be passed to the query.
+      query_specification: Query text that will be passed to API
+        alongside column_names, customizers and virtual columns.
+      args: Arguments that need to be passed to the query.
 
     Returns:
-        GarfReport with results of query execution.
+      GarfReport with results of query execution.
 
     Raises:
-        GarfExecutorException:
-            When customer_ids are not provided or API returned error.
+      GarfExecutorException:
+        When customer_ids are not provided or API returned error.
     """
     if not isinstance(query_specification, query_editor.QuerySpecification):
-      query_specification = query_editor.QuerySpecification(
+      query_specification = self.query_specification_builder(
         text=str(query_specification),
         args=args,
       )
@@ -103,3 +112,26 @@ class ApiReportFetcher:
     return report.GarfReport(
       results=parsed_response, column_names=query.column_names
     )
+
+
+class RestApiReportFetcher(ApiReportFetcher):
+  """Fetches data from an REST API endpoint.
+
+  Attributes:
+    api_client: Initialized RestApiClient.
+    parser: Type of parser to convert API response.
+  """
+
+  def __init__(
+    self,
+    endpoint: str,
+    parser: parsers.BaseParser = parsers.DictParser,
+  ) -> None:
+    """Instantiates RestApiReportFetcher.
+
+    Args:
+      endpoint: URL of API endpoint.
+      parser: Type of parser to convert API response.
+    """
+    self.api_client = api_clients.RestApiClient(endpoint)
+    self.parser = parser()

{garf_core-0.0.6.dist-info → garf_core-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: garf-core
-Version: 0.0.6
+Version: 0.0.8
 Summary: Abstracts fetching data from API based on provided SQL-like query.
 Author-email: "Google Inc. (gTech gPS CSE team)" <no-reply@google.com>
 License: Apache 2.0

garf_core-0.0.8.dist-info/RECORD

@@ -0,0 +1,13 @@
+garf_core/__init__.py,sha256=-2NI7Qm5ndrlcaz3itBBuFXOKu4anul2TdJro2G5oRo,598
+garf_core/api_clients.py,sha256=FKmwts_QBV_E4aTmrnsZtow0CmSExLibfeSkMSrudU0,2235
+garf_core/base_query.py,sha256=uEhKP56wcFtMiSwYOxoZ0q7OEvURGVOeRdpSYjxWLho,1201
+garf_core/exceptions.py,sha256=4qvNN-8GqbbVsrLKxJwBeX1FUtWpKbhUWyvm7anD3iE,1916
+garf_core/parsers.py,sha256=Y4wwpDXRZky7LBTBklvzk-aDNxgRL5RJg52rSdH5t9Q,2928
+garf_core/query_editor.py,sha256=EzwjFnbje-RcP9VwWQwzCpy8lipt4tph3Kv6q5Lly5o,15994
+garf_core/report.py,sha256=xlkZ44cdT_uGENuNeGwy69h7XLtwjdUjyPw3S1m-zc8,17852
+garf_core/report_fetcher.py,sha256=j1ewWoUvnEGM5vmWihFJ7vSVVEG1t2yzapKhylc-7eQ,4157
+garf_core-0.0.8.dist-info/METADATA,sha256=Ap-TAWp1wzDtXO6WPUBE1ajGnzQ_YMUIRiIhZbzLcm8,2334
+garf_core-0.0.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+garf_core-0.0.8.dist-info/entry_points.txt,sha256=ODxSZXwWEIXQAjRwBQsBJ6GYw8oPK6DYTo0oDVkKCpo,39
+garf_core-0.0.8.dist-info/top_level.txt,sha256=Gj-Zp7fM2turGut5vTJuo5xEcKfow7cTLX3y3WFfNgA,10
+garf_core-0.0.8.dist-info/RECORD,,

garf_core-0.0.8.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[garf]
+rest = garf_core.report_fetcher

garf_core-0.0.6.dist-info/RECORD

@@ -1,12 +0,0 @@
-garf_core/__init__.py,sha256=iynLwCh4JRTAmiLLrvX9e13zb6jgue4iR-IL7LOKZTw,598
-garf_core/api_clients.py,sha256=sBiOfPllrSewV6Ul4uaHhdmgTHQ_l5M7sZHW1z4kNik,1568
-garf_core/base_query.py,sha256=uEhKP56wcFtMiSwYOxoZ0q7OEvURGVOeRdpSYjxWLho,1201
-garf_core/exceptions.py,sha256=4qvNN-8GqbbVsrLKxJwBeX1FUtWpKbhUWyvm7anD3iE,1916
-garf_core/parsers.py,sha256=Y4wwpDXRZky7LBTBklvzk-aDNxgRL5RJg52rSdH5t9Q,2928
-garf_core/query_editor.py,sha256=EzwjFnbje-RcP9VwWQwzCpy8lipt4tph3Kv6q5Lly5o,15994
-garf_core/report.py,sha256=xlkZ44cdT_uGENuNeGwy69h7XLtwjdUjyPw3S1m-zc8,17852
-garf_core/report_fetcher.py,sha256=x6Hry0PfOWJ8Rwz-yjncj2xO8bDiwkyKrvtX-3VjLVk,3175
-garf_core-0.0.6.dist-info/METADATA,sha256=O-0SL-Nsbxh8WFIeQC5hqlpUh8ZGd9_lZ5FPC2CLGWw,2334
-garf_core-0.0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-garf_core-0.0.6.dist-info/top_level.txt,sha256=Gj-Zp7fM2turGut5vTJuo5xEcKfow7cTLX3y3WFfNgA,10
-garf_core-0.0.6.dist-info/RECORD,,