Nose
c019c00aed
Some checks failed
smoke-mm / Install & smoke test mm --help (push) Has been cancelled
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from typing import Any
|
|
|
|
from ProviderCore.base import Provider
|
|
from SYS.logger import log
|
|
|
|
|
|
class ZeroXZero(Provider):
|
|
"""File provider for 0x0.st."""
|
|
|
|
def upload(self, file_path: str, **kwargs: Any) -> str:
|
|
from API.HTTP import HTTPClient
|
|
from models import ProgressFileReader
|
|
|
|
if not os.path.exists(file_path):
|
|
raise FileNotFoundError(f"File not found: {file_path}")
|
|
|
|
try:
|
|
headers = {"User-Agent": "Medeia-Macina/1.0"}
|
|
with HTTPClient(headers=headers) as client:
|
|
with open(file_path, "rb") as handle:
|
|
try:
|
|
total = os.path.getsize(file_path)
|
|
except Exception:
|
|
total = None
|
|
wrapped = ProgressFileReader(handle, total_bytes=total, label="upload")
|
|
response = client.post("https://0x0.st", files={"file": wrapped})
|
|
|
|
if response.status_code == 200:
|
|
uploaded_url = response.text.strip()
|
|
try:
|
|
pipe_obj = kwargs.get("pipe_obj")
|
|
if pipe_obj is not None:
|
|
from Store import Store
|
|
|
|
Store(self.config, suppress_debug=True).try_add_url_for_pipe_object(
|
|
pipe_obj, uploaded_url
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
return uploaded_url
|
|
|
|
raise Exception(f"Upload failed: {response.status_code} - {response.text}")
|
|
|
|
except Exception as exc:
|
|
log(f"[0x0] Upload error: {exc}", file=sys.stderr)
|
|
raise
|
|
|
|
def validate(self) -> bool:
|
|
return True
|