mirror of
https://github.com/anasty17/mirror-leech-telegram-bot.git
synced 2025-01-08 12:07:33 +08:00
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
import os
|
|
import subprocess
|
|
import requests
|
|
import logging
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
if os.path.exists('log.txt'):
|
|
with open('log.txt', 'r+') as f:
|
|
f.truncate(0)
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[logging.FileHandler('log.txt'), logging.StreamHandler()],
|
|
level=logging.INFO)
|
|
|
|
CONFIG_FILE_URL = os.environ.get('CONFIG_FILE_URL', None)
|
|
try:
|
|
if len(CONFIG_FILE_URL) == 0:
|
|
raise TypeError
|
|
except TypeError:
|
|
CONFIG_FILE_URL = None
|
|
|
|
if CONFIG_FILE_URL is not None:
|
|
try:
|
|
res = requests.get(CONFIG_FILE_URL)
|
|
if res.status_code == 200:
|
|
with open('config.env', 'wb+') as f:
|
|
f.write(res.content)
|
|
f.close()
|
|
else:
|
|
logging.error(f"Failed to download config.env {res.status_code}")
|
|
except Exception as e:
|
|
logging.error(str(e))
|
|
|
|
load_dotenv('config.env', override=True)
|
|
|
|
UPSTREAM_REPO = os.environ.get('UPSTREAM_REPO', None)
|
|
try:
|
|
if len(UPSTREAM_REPO) == 0:
|
|
raise TypeError
|
|
except TypeError:
|
|
UPSTREAM_REPO = None
|
|
|
|
if UPSTREAM_REPO is not None:
|
|
if not os.path.exists('.git'):
|
|
subprocess.run([f"git init -q \
|
|
&& git config --global user.email e.anastayyar@gmail.com \
|
|
&& git config --global user.name mltb \
|
|
&& git add . \
|
|
&& git commit -sm update -q \
|
|
&& git remote add origin {UPSTREAM_REPO} \
|
|
&& git fetch origin -q \
|
|
&& git reset --hard origin/master -q"], shell=True)
|
|
else:
|
|
subprocess.run([f"git init -q \
|
|
&& git config --global user.email e.anastayyar@gmail.com \
|
|
&& git config --global user.name mltb \
|
|
&& git add . \
|
|
&& git commit -m update -q \
|
|
&& git remote rm origin \
|
|
&& git remote add origin {UPSTREAM_REPO} \
|
|
&& git fetch origin -q \
|
|
&& git reset --hard origin/master -q"], shell=True)
|
|
|