NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-08 08:38:16 -06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

14731 add user agent string, proxy settings

Browse Source
This commit is contained in:
Arthur Hanson 2024-07-25 15:38:50 +07:00
parent 6b16388c54
commit 54e4a9fd2b
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
netbox/core/plugins.py
View File

@ -12,6 +12,8 @@ from django.utils.translation import gettext_lazy as _
from netbox.plugins import PluginConfig from netbox.plugins import PluginConfig
from utilities.datetime import datetime_from_timestamp from utilities.datetime import datetime_from_timestamp
USER_AGENT_STRING = f'NetBox/{settings.RELEASE.version} {settings.RELEASE.edition}'
@dataclass @dataclass
class PluginAuthor: class PluginAuthor:
@ -98,13 +100,25 @@ def get_catalog_plugins():
def get_pages(): def get_pages():
# TODO: pagination is currently broken in API # TODO: pagination is currently broken in API
payload = {'page': '1', 'per_page': '50'} payload = {'page': '1', 'per_page': '50'}
first_page = session.get(settings.PLUGIN_CATALOG_URL, params=payload).json() first_page = session.get(
settings.PLUGIN_CATALOG_URL,
headers={'User-Agent': USER_AGENT_STRING},
proxies=settings.HTTP_PROXIES,
timeout=3,
params=payload
).json()
yield first_page yield first_page
num_pages = first_page['metadata']['pagination']['last_page'] num_pages = first_page['metadata']['pagination']['last_page']
for page in range(2, num_pages + 1): for page in range(2, num_pages + 1):
payload['page'] = page payload['page'] = page
next_page = session.get(settings.PLUGIN_CATALOG_URL, params=payload).json() next_page = session.get(
settings.PLUGIN_CATALOG_URL,
headers={'User-Agent': USER_AGENT_STRING},
proxies=settings.HTTP_PROXIES,
timeout=3,
params=payload
).json()
yield next_page yield next_page
for page in get_pages(): for page in get_pages():