/
update.py
executable file
·108 lines (92 loc) · 3.74 KB
/
update.py
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
import os
import time
import requests
# Global variables
config_version_path = "./config/version.cfg"
config_udfl_path = "./config/udfl"
tookie_osint_update_path = (
"https://raw.githubusercontent.com/alfredredbird/tookie-osint/main/config/version.cfg"
)
tookie_osint_install_path = "/tookie-osint/"
gitfile_location = "https://raw.githubusercontent.com/alfredredbird/tookie-osint/main/"
dl_count = 0
class Updater:
def __init__(self):
self.version = self.read_file(config_version_path)
self.udfl_list = self.read_file_lines(config_udfl_path)
def read_file(self, file_path):
with open(file_path, "r") as file:
return file.read().strip()
def read_file_lines(self, file_path):
with open(file_path, "r") as file:
return [line.strip() for line in file]
def save_file_from_url(self, url, destination):
try:
response = requests.get(url, allow_redirects=True)
os.makedirs(os.path.dirname(destination), exist_ok=True)
with open(destination, "wb") as file:
file.write(response.content)
return True
except Exception as e:
print(f"Error saving file from {url}: {e}")
return False
def update(self):
try:
remote_version = requests.get(tookie_osint_update_path).text.strip()
except requests.ConnectionError:
print("Failed to fetch updates. (-1)")
return
if remote_version != self.version:
self.process_update(remote_version)
else:
print("You're on the latest version!")
def process_update(self, remote_version):
print("Fetching Updates!")
self.delete_files()
self.save_file_from_url(gitfile_location + "config/udfl", config_udfl_path)
udfl = self.read_file_lines(config_udfl_path)
self.download_files(udfl)
self.verify_and_exec(udfl)
def delete_files(self):
if self.udfl_list is not None and isinstance(self.udfl_list, list):
for file_path in self.udfl_list:
if "update.py" not in file_path:
try:
os.remove(file_path)
except FileNotFoundError:
print(f"Skipping: {file_path}")
def download_files(self, file_list):
global dl_count
print("Downloading Files")
for file_path in file_list:
url = gitfile_location + file_path
if self.save_file_from_url(url, file_path):
dl_count += 1
self.progress_bar_manual(dl_count, len(file_list))
def verify_and_exec(self, file_list):
missing_files = [f for f in file_list if not os.path.exists(f)]
for file_path in missing_files:
self.save_file_from_url(gitfile_location + file_path, file_path)
if not missing_files:
print("Update Done!")
exec(open("brib.py").read())
def progress_bar_manual(self, current, total, length=40, fill="#"):
percent = (current / total) * 100
filled_length = int(length * percent / 100)
bar = fill * filled_length + "-" * (length - filled_length)
print(f"\rProgress: [{bar}] {percent:.2f}% Complete", end="\r", flush=True)
def reinstall(self):
print("Reinstalling.....!")
self.delete_files()
self.save_file_from_url(gitfile_location + "config/udfl", config_udfl_path)
self.download_files(self.udfl_list)
self.verify_and_exec(self.udfl_list)
# Usage of the class
choice = input("Update or reinstall? [U/r]: ").lower()
updater = Updater()
if choice == "u":
updater.update()
elif choice == "r":
updater.reinstall()
else:
print("Not sure what you meant... I'll ask later")