Skip to content

Commit

Permalink
Multithreaded fix
Browse files Browse the repository at this point in the history
  • Loading branch information
UltimaHoarder committed May 4, 2021
1 parent f1c6cdd commit 7f3362f
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions apis/api_helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,12 +47,12 @@ def multiprocessing(max_threads=None):


class session_manager():
def __init__(self, original_sessions=[], headers: dict = {}, session_rules=None, session_retry_rules=None) -> None:
def __init__(self, original_sessions=[], headers2: dict = {}, session_rules=None, session_retry_rules=None) -> None:
self.sessions = self.copy_sessions(original_sessions)
self.pool = multiprocessing()
self.max_threads = self.pool._processes
self.kill = False
self.headers = headers
self.headers = headers2
self.session_rules = session_rules
self.session_retry_rules = session_retry_rules

Expand Down Expand Up @@ -96,7 +96,8 @@ def process_links(link, session):
t1 = threading.Thread(target=do, args=[self])
t1.start()

def json_request(self, link: str, session: Union[Session] = None, Session=None, method="GET", stream=False, json_format=True, data={}, sleep=True, timeout=20, ignore_rules=False, headers: dict = {}) -> Any:
def json_request(self, link: str, session: Union[Session] = None, method="GET", stream=False, json_format=True, data={}, sleep=True, timeout=20, ignore_rules=False) -> Any:
headers = {}
if not session:
session = self.sessions[0]
if self.session_rules and not ignore_rules:
Expand Down

0 comments on commit 7f3362f

Please sign in to comment.