Compare commits

..

No commits in common. "53a262a1385cfdfe0e97ad4de183a90ec7992626" and "c79b3bfafd769ca678c5360e1d4c4a356e657ddf" have entirely different histories.

4 changed files with 6 additions and 11 deletions

View File

@ -20,7 +20,7 @@ def get_redis():
def redis_publish(rdb, item, item_project, item_type, item_subproject=None, item_category="x"): def redis_publish(rdb, item, item_project, item_type, item_subproject=None, item_category="x"):
item_project = item_project.replace(".", "-") item_project = item_project.replace(".", "-")
item_subproject = item_subproject.replace(".", "-") if item_subproject else None item_subproject = item_subproject.replace(".", "-")
item_source = item_project if not item_subproject else f"{item_project}.{item_subproject}" item_source = item_project if not item_subproject else f"{item_project}.{item_subproject}"
@ -35,16 +35,14 @@ def redis_publish(rdb, item, item_project, item_type, item_subproject=None, ite
rdb.lpush(routing_key, item) rdb.lpush(routing_key, item)
def get_web(session=None): def get_web():
ua = UserAgent() ua = UserAgent()
return Web( return Web(
session=session,
proxy=os.environ.get("PROXY", None), proxy=os.environ.get("PROXY", None),
rps=os.environ.get("RPS", 1), rps=os.environ.get("RPS", 1),
logger=stdout_logger, logger=stdout_logger,
cookie_file=os.environ.get("COOKIE_FILE", None), cookie_file=os.environ.get("COOKIE_FILE", None),
retry_codes=set(int(x) if x else None for x in os.environ.get("RETRY_CODES", "").split(",")), retry_codes=set(int(x) if x else None for x in os.environ.get("RETRY_CODES", "").split(",")),
retry_sleep=int(os.environ.get("RETRY_SLEEP", 0)),
ua=ua[os.environ.get("USER_AGENT")] if os.environ.get("USER_AGENT", None) is not None else None ua=ua[os.environ.get("USER_AGENT")] if os.environ.get("USER_AGENT", None) is not None else None
) )

View File

@ -4,14 +4,12 @@ import time
from threading import Lock from threading import Lock
import atexit import atexit
from time import sleep
import siphash import siphash
last_time_called = dict() last_time_called = dict()
def retry(attempts, callback=None, retry_sleep=0): def retry(attempts, callback=None):
def decorate(func): def decorate(func):
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
retries = attempts retries = attempts
@ -22,7 +20,6 @@ def retry(attempts, callback=None, retry_sleep=0):
if callback: if callback:
callback(e) callback(e)
retries -= 1 retries -= 1
sleep(retry_sleep)
return wrapper return wrapper
return decorate return decorate

View File

@ -112,7 +112,7 @@ def download_file(url, destination, session=None, headers=None, overwrite=False,
class Web: class Web:
def __init__(self, proxy=None, rps=1, retries=3, retry_sleep=0, logger=None, cookie_file=None, retry_codes=None, session=None, def __init__(self, proxy=None, rps=1, retries=3, logger=None, cookie_file=None, retry_codes=None, session=None,
ua=None): ua=None):
self._cookie_file = cookie_file self._cookie_file = cookie_file
self._proxy = proxy self._proxy = proxy
@ -140,7 +140,7 @@ class Web:
} }
@rate_limit(rps) @rate_limit(rps)
@retry(retries, callback=self._error_callback, retry_sleep=retry_sleep) @retry(retries, callback=self._error_callback)
def get(url, **kwargs): def get(url, **kwargs):
self._current_req = "GET", url, kwargs self._current_req = "GET", url, kwargs
r = self._session.get(url, **kwargs) r = self._session.get(url, **kwargs)

View File

@ -2,7 +2,7 @@ from setuptools import setup
setup( setup(
name="hexlib", name="hexlib",
version="1.37", version="1.34",
description="Misc utility methods", description="Misc utility methods",
author="simon987", author="simon987",
author_email="me@simon987.net", author_email="me@simon987.net",