TelegramTools/config.py at main · DilshanHarshajith/TelegramTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
"""
config.py  –  Central configuration for TelegramTools  (optional)
------------------------------------------------------------------
When present, every module imports this for credentials and settings.
When absent, each module falls back to its own inline resolver.
"""

import os
import sys
import getpass
from pathlib import Path


# ---------------------------------------------------------------------------
# Credential resolution: .env → env vars → interactive prompt
# ---------------------------------------------------------------------------

def _load_dotenv(path):
    vals = {}
    try:
        with open(path, encoding="utf-8") as f:
            for line in f:
                line = line.strip()
                if not line or line.startswith("#") or "=" not in line:
                    continue
                key, _, val = line.partition("=")
                vals[key.strip()] = val.strip().strip('"').strip("'")
    except OSError:
        pass
    return vals


def _resolve_credentials():
    api_id = api_hash = None
    session = "session"

    # 1. .env — project root (this file's dir), home dir, cwd
    _here = Path(__file__).resolve().parent
    for p in dict.fromkeys([_here / ".env", Path.home() / ".env", Path.cwd() / ".env"]):
        if p.is_file():
            d = _load_dotenv(p)
            api_id   = d.get("API_ID")   or api_id
            api_hash = d.get("API_HASH")  or api_hash
            session  = d.get("SESSION_NAME", session)
            if api_id and api_hash:
                print(f"[*] Credentials loaded from {p}")
                return int(api_id), api_hash, session

    # 2. Environment variables
    api_id   = os.getenv("API_ID")   or api_id
    api_hash = os.getenv("API_HASH") or api_hash
    if api_id and api_hash:
        return int(api_id), api_hash, session

    # 3. Interactive prompt
    print("\n[!] Telegram API credentials not found.")
    print("    Get yours at https://my.telegram.org/apps\n")
    while not api_id:
        raw = input("    API_ID  (numeric): ").strip()
        if raw.isdigit():
            api_id = raw
        else:
            print("    API_ID must be a number — please try again.")
    while not api_hash:
        raw = getpass.getpass("    API_HASH (hidden): ").strip()
        if raw:
            api_hash = raw
        else:
            print("    API_HASH cannot be empty — please try again.")

    if input("\n    Save to .env in current directory? [y/N] ").strip().lower() == "y":
        env_path = Path.cwd() / ".env"
        with open(env_path, "a", encoding="utf-8") as f:
            f.write(f"\nAPI_ID={api_id}\nAPI_HASH={api_hash}\n")
        print(f"    Saved to {env_path}\n")

    return int(api_id), api_hash, session


API_ID, API_HASH, SESSION_NAME = _resolve_credentials()

# ---------------------------------------------------------------------------
# Paths
# ---------------------------------------------------------------------------
DATA_DIR   = "data"
GROUP_FILE = os.path.join(DATA_DIR, "groups.txt")
OUTPUT_DIR = os.path.join(DATA_DIR, "output")

# ---------------------------------------------------------------------------
# Scraping defaults
# ---------------------------------------------------------------------------
DEFAULT_KEYWORD  = ""
DEFAULT_LIMIT    = 1000
REPLY_ITER_LIMIT = 500

# ---------------------------------------------------------------------------
# Output verbosity
# ---------------------------------------------------------------------------
VERBOSE  = True
INFO     = True
SUCCESS  = True
WARNING  = False
ERROR    = False
PROGRESS = True

# ---------------------------------------------------------------------------
# Module discovery filters  (main.py)
# ---------------------------------------------------------------------------
DISCOVER_IGNORE = ['!']
WEB_IGNORE      = ['#']

# ---------------------------------------------------------------------------
# Output path regex patterns  (main.py)
# ---------------------------------------------------------------------------
FILE_PATH_PATTERNS = [
    r'[Ss]aved to\s+([\w\-/\\:. ]+)',
    r'[Ee]xported to\s+([\w\-/\\:. ]+)',
    r'[Rr]esults for this group are in\s+([\w\-/\\:. ]+)',
    r'[Oo]utput(?:\s+path)?[:\s]+([\w\-/\\:. ]+)',
    r'[Pp]rocessing group[:\s]+([\w\-/\\:. ]+)',
    r'[Cc]ollecting (?:infrastructure|data) for[:\s]+([\w\-/\\:. ]+)',
]