exporter-ng: argparse removed, reformatted

This commit is contained in:
Piotr Dec 2025-04-30 00:12:16 +02:00
parent 2a8ea1b8f4
commit 0d22af2470
No known key found for this signature in database
GPG key ID: D3B5A5D0150D147A

View file

@ -1,16 +1,15 @@
import json
import os
import sys
import requests
import json
from timeit import default_timer
from dataclasses import dataclass
from datetime import datetime
import argparse
from time import sleep
from typing import List, Optional, Dict, Any
import requests
from dotenv import load_dotenv
from pathvalidate import sanitize_filename
from time import sleep
from dataclasses import dataclass
from typing import List, Optional, Dict, Any, Iterator, Tuple
from abc import ABC, abstractmethod
@dataclass
class SlackConfig:
@ -30,8 +29,10 @@ class SlackConfig:
except KeyError:
raise ValueError("Brak SLACK_USER_TOKEN w zmiennych środowiskowych")
class SlackAPI:
"""Klasa do komunikacji z API Slacka"""
def __init__(self, config: SlackConfig):
self.config = config
self.headers = {"Authorization": f"Bearer {config.user_token}"}
@ -130,6 +131,7 @@ class SlackAPI:
params["channel"] = channel
return self.paginated_get("files.list", params, "files")
@dataclass
class SlackUser:
"""Reprezentacja użytkownika Slack"""
@ -189,8 +191,10 @@ class SlackUser:
return ", ".join(parts)
class SlackChannel:
"""Reprezentacja kanału Slack"""
def __init__(self, data: Dict):
self.id = data["id"]
self.name = data.get("name", "")
@ -231,6 +235,7 @@ class SlackChannel:
return " ".join(parts)
@dataclass
class SlackFile:
"""Reprezentacja pliku Slack"""
@ -246,8 +251,10 @@ class SlackFile:
url_private=data.get("url_private", "")
)
class SlackMessage:
"""Reprezentacja wiadomości Slack"""
def __init__(self, data: Dict, users: Dict[str, SlackUser]):
self.timestamp = float(data["ts"])
self.text = data.get("text", "[no message content]")
@ -295,8 +302,10 @@ class SlackMessage:
return message + "\n\n" + "*" * 24 + "\n\n"
class SlackExporter:
"""Główna klasa eksportera"""
def __init__(self, config: SlackConfig, output_dir: str):
self.api = SlackAPI(config)
self.timestamp = datetime.now().strftime("%Y-%m-%d_%H%M%S")
@ -422,58 +431,7 @@ class SlackExporter:
print(f"Error downloading file {filename}: {e}. {attempts} attempts left.")
return self.download_file(filename, url, attempts - 1)
def main():
parser = argparse.ArgumentParser(description="Eksporter danych ze Slacka")
parser.add_argument("-o", help="Katalog wyjściowy (jeśli pusty, wyświetla na stdout)")
parser.add_argument("--lc", action="store_true", help="Lista wszystkich konwersacji")
parser.add_argument("--lu", action="store_true", help="Lista wszystkich użytkowników")
parser.add_argument("--json", action="store_true", help="Wynik w formacie JSON")
parser.add_argument("-c", action="store_true", help="Historia wszystkich dostępnych konwersacji")
parser.add_argument("--ch", help="Z -c, ogranicza eksport do podanego ID kanału")
parser.add_argument("--fr", help="Z -c, timestamp początku zakresu (Unix)")
parser.add_argument("--to", help="Z -c, timestamp końca zakresu (Unix)")
parser.add_argument("-r", action="store_true", help="Pobierz wątki ze wszystkich konwersacji")
parser.add_argument("--files", action="store_true", help="Pobierz wszystkie pliki")
args = parser.parse_args()
if args.files and not args.o:
print("Opcja --files wymaga określenia katalogu wyjściowego (-o)")
sys.exit(1)
try:
config = SlackConfig.from_env()
exporter = SlackExporter(config)
if args.o:
exporter.set_output_dir(args.o)
exporter.load_users()
exporter.load_channels()
if args.lc:
exporter.export_channel_list(args.json)
if args.lu:
exporter.export_user_list(args.json)
if args.c or args.r:
channel_ids = [args.ch] if args.ch else [ch.id for ch in exporter.channels]
for channel_id in channel_ids:
if args.c:
exporter.export_channel_history(channel_id, args.fr, args.to, args.json)
if args.r:
exporter.export_channel_replies(channel_id, args.fr, args.to, args.json)
if args.files and args.o:
# TODO: Implementacja pobierania plików
print("Funkcja pobierania plików jeszcze nie zaimplementowana")
except Exception as e:
print(f"Błąd: {e}")
sys.exit(1)
if __name__ == "__main__":
# main()
config = SlackConfig.from_env()
exporter = SlackExporter(config, 'out')