github/StreamingCommunity
1
1
Fork 0
mirror of https://github.com/Arrowar/StreamingCommunity.git synced 2025-06-07 12:05:35 +00:00
Code Issues 2 Packages Projects Releases Wiki Activity

Fix tmp dir for mac in user agent

Browse Source
This commit is contained in:
Ghost 2024-04-12 17:43:20 +02:00
parent ddcb3dfb71
commit f91d734b8e
1 changed files with 3 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
Src/Lib/Request/user_agent.py
View File

@ -6,6 +6,7 @@ import os
import random
import threading
import json
import tempfile
from typing import Dict, List
# Internal utilities
@ -47,33 +48,11 @@ def update_user_agents(browser_name: str, browser_user_agents: Dict[str, List[st
browser_user_agents[browser_name] = get_browser_user_agents_online(browser_name)
def create_temp_directory():
"""
Create the TEMP directory if it doesn't exist.
If the TEMP environment variable doesn't exist, this function creates
the TEMP directory in the user's local AppData folder.
Returns:
str: The path to the TEMP directory.
"""
# Get the TEMP directory from the environment variables
temp_dir = os.environ.get('TEMP')
# If TEMP directory is not set, create it in the local AppData folder
if temp_dir is None:
temp_dir = os.path.join(os.path.expanduser("~"), "AppData", "Local", "Temp")
os.makedirs(temp_dir, exist_ok=True)
return temp_dir
def create_or_update_user_agent_file() -> None:
"""
Create or update the user agent file with browser user agents.
"""
user_agent_file = os.path.join(os.environ.get('TEMP'), 'fake_user_agent.json')
user_agent_file = os.path.join(tempfile.gettempdir(), 'fake_user_agent.json')
if not os.path.exists(user_agent_file):
browser_user_agents: Dict[str, List[str]] = {}
@ -101,11 +80,8 @@ class UserAgentManager:
"""
def __init__(self):
# Create temp directory if not exist
temp_dir = create_temp_directory()
# Get path to temp file where save all user agents
self.user_agent_file = os.path.join(temp_dir, 'fake_user_agent.json')
self.user_agent_file = os.path.join(tempfile.gettempdir(), 'fake_user_agent.json')
# If file dont exist, creaet it
if not os.path.exists(self.user_agent_file):