Add function to list all entities of a specific type (#120)

Author: jasper-elimity

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max_line_length = 120

.github/workflows/pull_request.yml

@@ -6,6 +6,8 @@ jobs:
       - uses: actions/checkout@v3
       - run: /root/.local/bin/poetry install
       - run: /root/.local/bin/poetry run black --check .
+      - run: /root/.local/bin/poetry run flake8
+      - run: /root/.local/bin/poetry run isort -c .
       - run: /root/.local/bin/poetry run mypy .
       - run: /root/.local/bin/poetry run pydocstyle
 

.isort.cfg

@@ -0,0 +1,2 @@
+[settings]
+profile = black

poetry.lock

@@ -17,10 +17,10 @@ python-dateutil = "^2.8.2"
 requests = "^2.28.1"
 simplejson = "^3.17.6"
 typing-extensions = "^4.3.0"
+more-itertools = "^8.14.0"
 
 [tool.poetry.dev-dependencies]
 black = "^22.6.0"
-pydocstyle = "^6.1.1"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "^0.981"
@@ -31,4 +31,6 @@ pytest = "^7.1.3"
 hypothesis = "^6.54.6"
 jsonschema = "^4.16.0"
 types-jsonschema = "^4.16.1"
-
+isort = "^5.10.1"
+flake8 = "^5.0.4"
+pydocstyle = "^6.1.1"

pyproject.toml

@@ -5,6 +5,13 @@
 being defined in the local system timezone.
 """
 
+from elimity_insights_client._domain_graph_schema import (
+    AttributeType,
+    DomainGraphSchema,
+    EntityType,
+    RelationshipAttributeType,
+    Type,
+)
 from elimity_insights_client._elimity_insights_client import (
     AttributeAssignment,
     BooleanValue,
@@ -24,13 +31,6 @@
     TimeValue,
     Value,
 )
-from elimity_insights_client._domain_graph_schema import (
-    AttributeType,
-    DomainGraphSchema,
-    EntityType,
-    RelationshipAttributeType,
-    Type,
-)
 
 __all__ = [
     "AttributeAssignment",

src/elimity_insights_client/__init__.py

@@ -1,12 +1,12 @@
 from typing import List
 
-from typing_extensions import TypedDict, NotRequired
+from typing_extensions import NotRequired, TypedDict
 
 from elimity_insights_client._domain_graph_schema import (
-    DomainGraphSchema,
     AttributeType,
-    RelationshipAttributeType,
+    DomainGraphSchema,
     EntityType,
+    RelationshipAttributeType,
     Type,
 )
 from elimity_insights_client._util import map_list

src/elimity_insights_client/_decode_domain_graph_schema.py

@@ -2,17 +2,15 @@
 from datetime import datetime
 from enum import Enum, auto
 from itertools import chain
-from typing import Optional, Union, Iterable, Tuple, Dict
+from typing import Dict, Iterable, Optional, Tuple, Union
 from zlib import compressobj
 
-from requests import request, Response
+from requests import Response, request
 
 from elimity_insights_client._decode_domain_graph_schema import (
     decode_domain_graph_schema,
 )
-from elimity_insights_client._domain_graph_schema import (
-    DomainGraphSchema,
-)
+from elimity_insights_client._domain_graph_schema import DomainGraphSchema
 from elimity_insights_client._util import encode_datetime, encoder
 
 

src/elimity_insights_client/_elimity_insights_client.py

@@ -1,9 +1,9 @@
 from datetime import datetime
-from simplejson import JSONEncoder
-from typing import TypeVar, Callable, List
+from typing import Callable, List, TypeVar
 
 from dateutil.tz import tzlocal
 from dateutil.utils import default_tzinfo
+from simplejson import JSONEncoder
 
 encoder = JSONEncoder(iterable_as_array=True)
 local_timezone = tzlocal()

src/elimity_insights_client/_util.py

@@ -1,19 +1,19 @@
 """Endpoints for API interactions with an Elimity Insights server."""
 
 from dataclasses import dataclass
-from typing import List, TypeVar, cast, Type, Optional
+from typing import List, Optional, Type, TypeVar, cast
 
 from requests import request
 
 from elimity_insights_client._util import encoder, map_list
 from elimity_insights_client.api._decode_query_results_page import (
-    decode_query_results_page,
     QueryResultsPageDict,
+    decode_query_results_page,
 )
 from elimity_insights_client.api._decode_source import SourceDict, decode_source
+from elimity_insights_client.api._encode_query import encode_query
 from elimity_insights_client.api.query import Query
 from elimity_insights_client.api.query_results_page import QueryResultsPage
-from elimity_insights_client.api._encode_query import encode_query
 from elimity_insights_client.api.source import Source