schihei/yolo-ai-cmdbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removed all pylint warnings

Browse Source
This commit is contained in:
Heiko Joerg Schick
2023-09-05 21:11:23 +02:00
parent 838d55563b
commit bc1e997708
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
yolo.py
+10 -12
View File
@@ -181,7 +181,6 @@ def prompt_user_input(config, response):
Todo Todo
""" """
print("Command: " + colored(response, 'blue')) print("Command: " + colored(response, 'blue'))
#print(config["safety"])
if config["safety"]: if config["safety"]:
prompt_text = "Execute command? [Y]es [n]o [m]odify [c]opy to clipboard ==> " prompt_text = "Execute command? [Y]es [n]o [m]odify [c]opy to clipboard ==> "
@@ -192,32 +191,31 @@ def prompt_user_input(config, response):
print(prompt_text, end = '') print(prompt_text, end = '')
user_input = input() user_input = input()
else:
user_input = "Y"
return user_input return user_input
if not config["safety"]: def evaluate_input(config, shell, user_input, command):
return "Y"
def evaluate_input(user_input, command):
""" """
Todo Todo
""" """
if user_input.upper() == "Y" or user_input == "": if user_input.upper() == "Y" or user_input == "":
if shell == "powershell.exe": if shell == "powershell.exe":
subprocess.run([shell, "/c", command], shell=False) subprocess.run([shell, "/c", command], shell=False, check=True)
else: else:
# Unix: /bin/bash /bin/zsh: uses -c both Ubuntu and macOS should work, others might not # Unix: /bin/bash /bin/zsh: uses -c both Ubuntu and macOS should work, others might not
subprocess.run([shell, "-c", command], shell=False) subprocess.run([shell, "-c", command], shell=False, check=True)
if user_input.upper() == "M": if user_input.upper() == "M":
print("Modify prompt: ", end = '') print("Modify prompt: ", end = '')
modded_query = input() modded_query = input()
modded_response = call_open_ai(modded_query) modded_response = call_open_ai(config, shell, modded_query)
check_for_issue(modded_response) check_for_issue(modded_response)
check_for_markdown(modded_response) check_for_markdown(modded_response)
modded_user_input = prompt_user_input(modded_response) modded_user_input = prompt_user_input(config, modded_response)
print() print()
evaluate_input(modded_user_input, modded_response) evaluate_input(config, shell, modded_user_input, modded_response)
if user_input.upper() == "C": if user_input.upper() == "C":
if os.name == "posix" and missing_posix_display(): if os.name == "posix" and missing_posix_display():
@@ -265,7 +263,7 @@ def main():
check_for_markdown(res_command) check_for_markdown(res_command)
user_input = prompt_user_input(config, res_command) user_input = prompt_user_input(config, res_command)
print() print()
evaluate_input(user_input, res_command) evaluate_input(config, shell, user_input, res_command)
if __name__ == "__main__": if __name__ == "__main__":
main() main()