github/gibMacOS
4
0
Fork 0
mirror of https://github.com/corpnewt/gibMacOS.git synced 2024-04-21 12:31:48 +00:00
Code Issues Releases Wiki Activity

Update utils.py, plist.py from pymodules

Browse Source
This commit is contained in:
CorpNewt
2020-04-29 15:35:46 -05:00
parent 9a60b44428
commit cce8677e57
1 changed files with 255 additions and 255 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Scripts/utils.py
Regular → Executable
+255 -255
View File
@@ -1,255 +1,255 @@
import sys, os, time, re, json, datetime, ctypes, subprocess import sys, os, time, re, json, datetime, ctypes, subprocess
if os.name == "nt": if os.name == "nt":
# Windows # Windows
import msvcrt import msvcrt
else: else:
# Not Windows \o/ # Not Windows \o/
import select import select
class Utils: class Utils:
def __init__(self, name = "Python Script"): def __init__(self, name = "Python Script"):
self.name = name self.name = name
# Init our colors before we need to print anything # Init our colors before we need to print anything
cwd = os.getcwd() cwd = os.getcwd()
os.chdir(os.path.dirname(os.path.realpath(__file__))) os.chdir(os.path.dirname(os.path.realpath(__file__)))
if os.path.exists("colors.json"): if os.path.exists("colors.json"):
self.colors_dict = json.load(open("colors.json")) self.colors_dict = json.load(open("colors.json"))
else: else:
self.colors_dict = {} self.colors_dict = {}
os.chdir(cwd) os.chdir(cwd)
def check_admin(self): def check_admin(self):
# Returns whether or not we're admin # Returns whether or not we're admin
try: try:
is_admin = os.getuid() == 0 is_admin = os.getuid() == 0
except AttributeError: except AttributeError:
is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0 is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
return is_admin return is_admin
def elevate(self, file): def elevate(self, file):
# Runs the passed file as admin # Runs the passed file as admin
if self.check_admin(): if self.check_admin():
return return
if os.name == "nt": if os.name == "nt":
ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, file, None, 1) ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, file, None, 1)
else: else:
try: try:
p = subprocess.Popen(["which", "sudo"], stdout=subprocess.PIPE, stderr=subprocess.PIPE) p = subprocess.Popen(["which", "sudo"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
c = p.communicate()[0].decode("utf-8", "ignore").replace("\n", "") c = p.communicate()[0].decode("utf-8", "ignore").replace("\n", "")
os.execv(c, [ sys.executable, 'python'] + sys.argv) os.execv(c, [ sys.executable, 'python'] + sys.argv)
except: except:
exit(1) exit(1)
def compare_versions(self, vers1, vers2, **kwargs): def compare_versions(self, vers1, vers2, **kwargs):
# Helper method to compare ##.## strings # Helper method to compare ##.## strings
# #
# vers1 < vers2 = True # vers1 < vers2 = True
# vers1 = vers2 = None # vers1 = vers2 = None
# vers1 > vers2 = False # vers1 > vers2 = False
# Sanitize the pads # Sanitize the pads
pad = str(kwargs.get("pad", "")) pad = str(kwargs.get("pad", ""))
sep = str(kwargs.get("separator", ".")) sep = str(kwargs.get("separator", "."))
ignore_case = kwargs.get("ignore_case", True) ignore_case = kwargs.get("ignore_case", True)
# Cast as strings # Cast as strings
vers1 = str(vers1) vers1 = str(vers1)
vers2 = str(vers2) vers2 = str(vers2)
if ignore_case: if ignore_case:
vers1 = vers1.lower() vers1 = vers1.lower()
vers2 = vers2.lower() vers2 = vers2.lower()
# Split and pad lists # Split and pad lists
v1_parts, v2_parts = self.pad_length(vers1.split(sep), vers2.split(sep)) v1_parts, v2_parts = self.pad_length(vers1.split(sep), vers2.split(sep))
# Iterate and compare # Iterate and compare
for i in range(len(v1_parts)): for i in range(len(v1_parts)):
# Remove non-numeric # Remove non-numeric
v1 = ''.join(c.lower() for c in v1_parts[i] if c.isalnum()) v1 = ''.join(c.lower() for c in v1_parts[i] if c.isalnum())
v2 = ''.join(c.lower() for c in v2_parts[i] if c.isalnum()) v2 = ''.join(c.lower() for c in v2_parts[i] if c.isalnum())
# Equalize the lengths # Equalize the lengths
v1, v2 = self.pad_length(v1, v2) v1, v2 = self.pad_length(v1, v2)
# Compare # Compare
if str(v1) < str(v2): if str(v1) < str(v2):
return True return True
elif str(v1) > str(v2): elif str(v1) > str(v2):
return False return False
# Never differed - return None, must be equal # Never differed - return None, must be equal
return None return None
def pad_length(self, var1, var2, pad = "0"): def pad_length(self, var1, var2, pad = "0"):
# Pads the vars on the left side to make them equal length # Pads the vars on the left side to make them equal length
pad = "0" if len(str(pad)) < 1 else str(pad)[0] pad = "0" if len(str(pad)) < 1 else str(pad)[0]
if not type(var1) == type(var2): if not type(var1) == type(var2):
# Type mismatch! Just return what we got # Type mismatch! Just return what we got
return (var1, var2) return (var1, var2)
if len(var1) < len(var2): if len(var1) < len(var2):
if type(var1) is list: if type(var1) is list:
var1.extend([str(pad) for x in range(len(var2) - len(var1))]) var1.extend([str(pad) for x in range(len(var2) - len(var1))])
else: else:
var1 = "{}{}".format((pad*(len(var2)-len(var1))), var1) var1 = "{}{}".format((pad*(len(var2)-len(var1))), var1)
elif len(var2) < len(var1): elif len(var2) < len(var1):
if type(var2) is list: if type(var2) is list:
var2.extend([str(pad) for x in range(len(var1) - len(var2))]) var2.extend([str(pad) for x in range(len(var1) - len(var2))])
else: else:
var2 = "{}{}".format((pad*(len(var1)-len(var2))), var2) var2 = "{}{}".format((pad*(len(var1)-len(var2))), var2)
return (var1, var2) return (var1, var2)
def check_path(self, path): def check_path(self, path):
# Let's loop until we either get a working path, or no changes # Let's loop until we either get a working path, or no changes
test_path = path test_path = path
last_path = None last_path = None
while True: while True:
# Bail if we've looped at least once and the path didn't change # Bail if we've looped at least once and the path didn't change
if last_path != None and last_path == test_path: return None if last_path != None and last_path == test_path: return None
last_path = test_path last_path = test_path
# Check if we stripped everything out # Check if we stripped everything out
if not len(test_path): return None if not len(test_path): return None
# Check if we have a valid path # Check if we have a valid path
if os.path.exists(test_path): if os.path.exists(test_path):
return os.path.abspath(test_path) return os.path.abspath(test_path)
# Check for quotes # Check for quotes
if test_path[0] == test_path[-1] and test_path[0] in ('"',"'"): if test_path[0] == test_path[-1] and test_path[0] in ('"',"'"):
test_path = test_path[1:-1] test_path = test_path[1:-1]
continue continue
# Check for a tilde and expand if needed # Check for a tilde and expand if needed
if test_path[0] == "~": if test_path[0] == "~":
tilde_expanded = os.path.expanduser(test_path) tilde_expanded = os.path.expanduser(test_path)
if tilde_expanded != test_path: if tilde_expanded != test_path:
# Got a change # Got a change
test_path = tilde_expanded test_path = tilde_expanded
continue continue
# Let's check for spaces - strip from the left first, then the right # Let's check for spaces - strip from the left first, then the right
if test_path[0] in (" ","\t"): if test_path[0] in (" ","\t"):
test_path = test_path[1:] test_path = test_path[1:]
continue continue
if test_path[-1] in (" ","\t"): if test_path[-1] in (" ","\t"):
test_path = test_path[:-1] test_path = test_path[:-1]
continue continue
# Maybe we have escapes to handle? # Maybe we have escapes to handle?
test_path = "\\".join([x.replace("\\", "") for x in test_path.split("\\\\")]) test_path = "\\".join([x.replace("\\", "") for x in test_path.split("\\\\")])
def grab(self, prompt, **kwargs): def grab(self, prompt, **kwargs):
# Takes a prompt, a default, and a timeout and shows it with that timeout # Takes a prompt, a default, and a timeout and shows it with that timeout
# returning the result # returning the result
timeout = kwargs.get("timeout", 0) timeout = kwargs.get("timeout", 0)
default = kwargs.get("default", None) default = kwargs.get("default", None)
# If we don't have a timeout - then skip the timed sections # If we don't have a timeout - then skip the timed sections
if timeout <= 0: if timeout <= 0:
if sys.version_info >= (3, 0): if sys.version_info >= (3, 0):
return input(prompt) return input(prompt)
else: else:
return str(raw_input(prompt)) return str(raw_input(prompt))
# Write our prompt # Write our prompt
sys.stdout.write(prompt) sys.stdout.write(prompt)
sys.stdout.flush() sys.stdout.flush()
if os.name == "nt": if os.name == "nt":
start_time = time.time() start_time = time.time()
i = '' i = ''
while True: while True:
if msvcrt.kbhit(): if msvcrt.kbhit():
c = msvcrt.getche() c = msvcrt.getche()
if ord(c) == 13: # enter_key if ord(c) == 13: # enter_key
break break
elif ord(c) >= 32: #space_char elif ord(c) >= 32: #space_char
i += c i += c
if len(i) == 0 and (time.time() - start_time) > timeout: if len(i) == 0 and (time.time() - start_time) > timeout:
break break
else: else:
i, o, e = select.select( [sys.stdin], [], [], timeout ) i, o, e = select.select( [sys.stdin], [], [], timeout )
if i: if i:
i = sys.stdin.readline().strip() i = sys.stdin.readline().strip()
print('') # needed to move to next line print('') # needed to move to next line
if len(i) > 0: if len(i) > 0:
return i return i
else: else:
return default return default
def cls(self): def cls(self):
os.system('cls' if os.name=='nt' else 'clear') os.system('cls' if os.name=='nt' else 'clear')
def cprint(self, message, **kwargs): def cprint(self, message, **kwargs):
strip_colors = kwargs.get("strip_colors", False) strip_colors = kwargs.get("strip_colors", False)
if os.name == "nt": if os.name == "nt":
strip_colors = True strip_colors = True
reset = u"\u001b[0m" reset = u"\u001b[0m"
# Requires sys import # Requires sys import
for c in self.colors: for c in self.colors:
if strip_colors: if strip_colors:
message = message.replace(c["find"], "") message = message.replace(c["find"], "")
else: else:
message = message.replace(c["find"], c["replace"]) message = message.replace(c["find"], c["replace"])
if strip_colors: if strip_colors:
return message return message
sys.stdout.write(message) sys.stdout.write(message)
print(reset) print(reset)
# Needs work to resize the string if color chars exist # Needs work to resize the string if color chars exist
'''# Header drawing method '''# Header drawing method
def head(self, text = None, width = 55): def head(self, text = None, width = 55):
if text == None: if text == None:
text = self.name text = self.name
self.cls() self.cls()
print(" {}".format("#"*width)) print(" {}".format("#"*width))
len_text = self.cprint(text, strip_colors=True) len_text = self.cprint(text, strip_colors=True)
mid_len = int(round(width/2-len(len_text)/2)-2) mid_len = int(round(width/2-len(len_text)/2)-2)
middle = " #{}{}{}#".format(" "*mid_len, len_text, " "*((width - mid_len - len(len_text))-2)) middle = " #{}{}{}#".format(" "*mid_len, len_text, " "*((width - mid_len - len(len_text))-2))
if len(middle) > width+1: if len(middle) > width+1:
# Get the difference # Get the difference
di = len(middle) - width di = len(middle) - width
# Add the padding for the ...# # Add the padding for the ...#
di += 3 di += 3
# Trim the string # Trim the string
middle = middle[:-di] middle = middle[:-di]
newlen = len(middle) newlen = len(middle)
middle += "...#" middle += "...#"
find_list = [ c["find"] for c in self.colors ] find_list = [ c["find"] for c in self.colors ]
# Translate colored string to len # Translate colored string to len
middle = middle.replace(len_text, text + self.rt_color) # always reset just in case middle = middle.replace(len_text, text + self.rt_color) # always reset just in case
self.cprint(middle) self.cprint(middle)
print("#"*width)''' print("#"*width)'''
# Header drawing method # Header drawing method
def head(self, text = None, width = 55): def head(self, text = None, width = 55):
if text == None: if text == None:
text = self.name text = self.name
self.cls() self.cls()
print(" {}".format("#"*width)) print(" {}".format("#"*width))
mid_len = int(round(width/2-len(text)/2)-2) mid_len = int(round(width/2-len(text)/2)-2)
middle = " #{}{}{}#".format(" "*mid_len, text, " "*((width - mid_len - len(text))-2)) middle = " #{}{}{}#".format(" "*mid_len, text, " "*((width - mid_len - len(text))-2))
if len(middle) > width+1: if len(middle) > width+1:
# Get the difference # Get the difference
di = len(middle) - width di = len(middle) - width
# Add the padding for the ...# # Add the padding for the ...#
di += 3 di += 3
# Trim the string # Trim the string
middle = middle[:-di] + "...#" middle = middle[:-di] + "...#"
print(middle) print(middle)
print("#"*width) print("#"*width)
def resize(self, width, height): def resize(self, width, height):
print('\033[8;{};{}t'.format(height, width)) print('\033[8;{};{}t'.format(height, width))
def custom_quit(self): def custom_quit(self):
self.head() self.head()
print("by CorpNewt\n") print("by CorpNewt\n")
print("Thanks for testing it out, for bugs/comments/complaints") print("Thanks for testing it out, for bugs/comments/complaints")
print("send me a message on Reddit, or check out my GitHub:\n") print("send me a message on Reddit, or check out my GitHub:\n")
print("www.reddit.com/u/corpnewt") print("www.reddit.com/u/corpnewt")
print("www.github.com/corpnewt\n") print("www.github.com/corpnewt\n")
# Get the time and wish them a good morning, afternoon, evening, and night # Get the time and wish them a good morning, afternoon, evening, and night
hr = datetime.datetime.now().time().hour hr = datetime.datetime.now().time().hour
if 3 < hr < 12: if hr > 3 and hr < 12:
print("Have a nice morning!\n\n") print("Have a nice morning!\n\n")
elif 12 <= hr < 17: elif hr >= 12 and hr < 17:
print("Have a nice afternoon!\n\n") print("Have a nice afternoon!\n\n")
elif 17 <= hr < 21: elif hr >= 17 and hr < 21:
print("Have a nice evening!\n\n") print("Have a nice evening!\n\n")
else: else:
print("Have a nice night!\n\n") print("Have a nice night!\n\n")
exit(0) exit(0)