Skip to content

Commit 9bb7e8c

Browse files
committed
IDEV-2020: Use typing.Generator.
1 parent 23f3fbf commit 9bb7e8c

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

domaintools/api.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from datetime import datetime, timedelta, timezone
22
from hashlib import sha1, sha256, md5
33
from hmac import new as hmac
4-
from types import GeneratorType
4+
from typing import Generator
55

66
import re
77

@@ -1068,7 +1068,7 @@ def iris_detect_ignored_domains(
10681068
**kwargs,
10691069
)
10701070

1071-
def nod(self, **kwargs) -> GeneratorType:
1071+
def nod(self, **kwargs) -> Generator:
10721072
"""Returns back list of the newly observed domains feed"""
10731073
validate_feeds_parameters(kwargs)
10741074
endpoint = kwargs.pop("endpoint", Endpoint.FEED.value)
@@ -1085,7 +1085,7 @@ def nod(self, **kwargs) -> GeneratorType:
10851085
**kwargs,
10861086
)
10871087

1088-
def nad(self, **kwargs) -> GeneratorType:
1088+
def nad(self, **kwargs) -> Generator:
10891089
"""Returns back list of the newly active domains feed"""
10901090
validate_feeds_parameters(kwargs)
10911091
endpoint = kwargs.pop("endpoint", Endpoint.FEED.value)
@@ -1102,7 +1102,7 @@ def nad(self, **kwargs) -> GeneratorType:
11021102
**kwargs,
11031103
)
11041104

1105-
def domainrdap(self, **kwargs) -> GeneratorType:
1105+
def domainrdap(self, **kwargs) -> Generator:
11061106
"""Returns changes to global domain registration information, populated by the Registration Data Access Protocol (RDAP)"""
11071107
validate_feeds_parameters(kwargs)
11081108
endpoint = kwargs.pop("endpoint", Endpoint.FEED.value)
@@ -1116,7 +1116,7 @@ def domainrdap(self, **kwargs) -> GeneratorType:
11161116
**kwargs,
11171117
)
11181118

1119-
def domaindiscovery(self, **kwargs) -> GeneratorType:
1119+
def domaindiscovery(self, **kwargs) -> Generator:
11201120
"""Returns new domains as they are either discovered in domain registration information, observed by our global sensor network, or reported by trusted third parties"""
11211121
validate_feeds_parameters(kwargs)
11221122
endpoint = kwargs.pop("endpoint", Endpoint.FEED.value)

domaintools/results.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from ordereddict import OrderedDict
1212

1313
from itertools import zip_longest
14-
from types import GeneratorType
14+
from typing import Generator
1515

1616
from domaintools_async import AsyncResults as Results
1717

@@ -148,7 +148,7 @@ def flattened(self):
148148
class FeedsResults(Results):
149149
"""Returns the generator for feeds results"""
150150

151-
def response(self) -> GeneratorType:
151+
def response(self) -> Generator:
152152
status_code = None
153153
while status_code != 200:
154154
resp_data = super().response()

0 commit comments

Comments
 (0)