Friendfeng
b3b036429a
renamed: FFAI/__pycache__/catch.cpython-313.pyc -> FFAICilent/__pycache__/catch.cpython-313.pyc renamed: FFAI/__pycache__/crawlers.cpython-313.pyc -> FFAICilent/__pycache__/crawlers.cpython-313.pyc renamed: FFAI/__pycache__/crawlers_core.cpython-313.pyc -> FFAICilent/__pycache__/crawlers_core.cpython-313.pyc renamed: FFAI/analyzer.py -> FFAICilent/analyzer.py renamed: FFAI/catch.py -> FFAICilent/catch.py new file: FFAICilent/cloud.py new file: FFAICilent/config/config.ini new file: FFAICilent/config/configloder.py renamed: FFAI/crawlers.py -> FFAICilent/crawlers.py renamed: FFAI/crawlers_core.py -> FFAICilent/crawlers_core.py new file: FFAICilent/local.py new file: FFAICilent/logger.py renamed: FFAI/main.py -> FFAICilent/main.py new file: FFAICilent/manger.py
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import requests
|
|
from ..security import decrypt_secret
|
|
from logger import log
|
|
|
|
class CloudConnector:
|
|
def __init__(self, config):
|
|
self.endpoint = config['Cloud']['endpoint']
|
|
self.timeout = int(config['Deployment']['cloud_timeout'])
|
|
self.retry = int(config['Cloud']['retry_times'])
|
|
self.token = decrypt_secret('api_token')
|
|
|
|
def _make_request(self, data):
|
|
headers = {
|
|
"Authorization": f"Bearer {self.token}",
|
|
"Content-Type": "application/json"
|
|
}
|
|
try:
|
|
resp = requests.post(
|
|
self.endpoint,
|
|
json=data,
|
|
headers=headers,
|
|
timeout=self.timeout
|
|
)
|
|
resp.raise_for_status()
|
|
return resp.json()
|
|
except requests.exceptions.RequestException as e:
|
|
log.error(f"Cloud request failed: {str(e)}")
|
|
raise ConnectionError("Cloud service unavailable")
|
|
|
|
def execute(self, command):
|
|
for attempt in range(self.retry):
|
|
try:
|
|
return self._make_request({"command": command})
|
|
except ConnectionError:
|
|
if attempt == self.retry - 1:
|
|
raise
|
|
log.warning(f"Retrying... ({attempt + 1}/{self.retry})") |