Repository navigation
Expand file tree
/
Copy pathuninstaller.py
More file actions
executable file
·140 lines (114 loc) · 4.84 KB
/
Copy pathuninstaller.py
File metadata and controls
executable file
·140 lines (114 loc) · 4.84 KB
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/usr/bin/env python3
import ctypes
import os
import platform
import subprocess
import sys
from shutil import rmtree, which
import colorama
# Platform indepent way to check if user is admin
def is_user_admin():
"""
Check if the script is being run as root/admin
Return False if privileges cannot be determined
"""
if platform.system() == 'Windows':
try:
return ctypes.windll.shell32.IsUserAnAdmin() == 1
except WindowsError:
return False
else:
try:
return os.getuid() == 0
except os.Error:
return False
# Uninstall Pip packages in a platform independent way
def uninstall_pip_packages():
"""Remove pip packages installed by pygoat"""
print(colorama.Back.CYAN + colorama.Style.BRIGHT + "[+] Uninstalling Pip packages!" + colorama.Style.RESET_ALL)
try:
# It is important to upgrade pip first to avoid environment errors
if (platform.system != 'Windows'):
pip_v = "pip3" if (which('pip3') is not None) else "pip"
subprocess.run([pip_v,
"install",
"--upgrade",
"pip"],
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL)
subprocess.check_call([sys.executable,
"-m",
"pip",
"uninstall",
"-yr",
"requirements.txt"])
except subprocess.CalledProcessError:
print(colorama.Fore.RED + colorama.Style.BRIGHT + "[!] Failed to uninstall pip packages" + colorama.Style.RESET_ALL)
# Uninstall PIP
def uninstall_pip():
"""Remove Pip"""
print(colorama.Back.RED + colorama.Style.BRIGHT + "[+] Uninstalling Pip!" + colorama.Style.RESET_ALL)
try:
subprocess.check_call([sys.executable,
"-m",
"pip",
"uninstall",
"-y",
"pip"])
except subprocess.CalledProcessError:
print(colorama.Fore.RED + colorama.Style.BRIGHT + "[!] Failed to uninstall pip" + colorama.Style.RESET_ALL)
# Remove pygoat
def remove_pygoat():
"""Remove pygoat files"""
cwd = os.getcwd()
print(colorama.Back.RED + colorama.Style.BRIGHT + f"All files in {cwd} will be deleted!" + colorama.Style.RESET_ALL)
for item in os.listdir(cwd):
if platform.system() == 'Windows':
filename = cwd + '\\' + item
else:
filename = cwd + '/' + item
if(os.path.isfile(filename)):
try:
print("[!] Deleted: " + colorama.Fore.RED + colorama.Style.BRIGHT + filename + colorama.Style.RESET_ALL)
os.remove(filename)
except os.Error:
print(colorama.Fore.RED + colorama.Style.BRIGHT + f"[!] Failed To remove: {filename}" + colorama.Style.RESET_ALL)
pass
if(os.path.isdir(filename)):
print("[!] Deleted: " + colorama.Fore.RED + colorama.Style.BRIGHT + filename + colorama.Style.RESET_ALL)
rmtree(filename, ignore_errors=True)
def main():
colorama.init()
# Check if program is being run as admin
# However, you need admin privileges only if you are not in a venv
if(not is_user_admin() and sys.prefix == sys.base_prefix):
print(colorama.Fore.RED + colorama.Style.BRIGHT + "[!] This script must be run as root!" + colorama.Style.RESET_ALL)
sys.exit(-1)
# Remove pip packages
uninstall_pip_packages()
# Remove pip
choice = input("Uninstall pip? (y/N) ")
if (choice.upper() == 'Y' or choice.upper() == 'YES'):
uninstall_pip()
else:
print(colorama.Back.CYAN + colorama.Style.BRIGHT + "[+] Pip has been kept intact" + colorama.Style.RESET_ALL)
# Remove pygoat files
choice = input(
"Would you like to remove all pygoat directories and files? (y/N) "
)
if (choice.upper() == 'Y' or choice.upper() == 'YES'):
remove_pygoat()
choice2 = input(f"Remove {os.getcwd()}? (y/N) ")
if (choice2.upper() == 'Y' or choice2.upper() == 'YES'):
try:
rmtree(os.getcwd(), ignore_errors=True)
print(colorama.Back.RED + colorama.Style.BRIGHT + f"[+] {os.getcwd()} has been removed" + colorama.Style.RESET_ALL)
except FileNotFoundError:
pass
else:
print(colorama.Back.CYAN + colorama.Style.BRIGHT + f"[+] {os.getcwd()} has been kept intact" + colorama.Style.RESET_ALL)
print(colorama.Back.RED + colorama.Style.BRIGHT + "Uninstallations Done!" + colorama.Style.RESET_ALL)
# Restore output streams to their original values
colorama.deinit()
if __name__ == '__main__':
main()