laramies/theHarvester
1
1
Fork 0
mirror of https://github.com/laramies/theHarvester.git synced 2024-09-20 15:26:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

#1383 fix githubcode.py

Browse source
This commit is contained in:
yoonthegoon 2023-07-24 12:59:01 -04:00
parent b7aba8ed7f
commit 8468e4767f
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
theHarvester/discovery/githubcode.py
View file

@ -17,7 +17,7 @@ class RetryResult(NamedTuple):
class SuccessResult(NamedTuple):
fragments: List[str]
next_page: Union[int, None]
last_page: int
last_page: Union[int, None]
class ErrorResult(NamedTuple):
@ -32,7 +32,7 @@ def __init__(self, word, limit) -> None:
self.server = "api.github.com"
self.limit = limit
self.counter: int = 0
self.page: int = 1
self.page: Union[int, None] = 1
self.key = Core.github_key()
# If you don't have a personal access token, GitHub narrows your search capabilities significantly
# rate limits you more severely
@ -71,7 +71,6 @@ async def handle_response(
if status == 200:
results = await self.fragments_from_response(json_data)
next_page = await self.page_from_response("next", links)
# TODO: figure out what int is last page
last_page = await self.page_from_response("last", links)
return SuccessResult(results, next_page, last_page)
elif status == 429 or status == 403:
@ -105,7 +104,7 @@ async def do_search(self, page: int) -> Tuple[str, dict, int, Any]:
return await resp.text(), await resp.json(), resp.status, resp.links
@staticmethod
async def next_page_or_end(result: SuccessResult) -> int:
async def next_page_or_end(result: SuccessResult) -> Union[int, None]:
if result.next_page is not None:
return result.next_page
else: