accounting/bench
2
0
Fork 0
mirror of https://github.com/frappe/bench.git synced 2025-01-09 16:36:25 +00:00
Code Issues Releases Activity

chore: rename print function to log

Browse Source
This commit is contained in:
Gavin D'souza 2019-12-19 14:38:46 +05:30
parent bc190a6d3b
commit 07d23d6911
1 changed files with 16 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
playbooks/install.py
View File

@ -1,6 +1,5 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import os, sys, subprocess, getpass, json, multiprocessing, shutil, platform, warnings import os, sys, subprocess, getpass, json, multiprocessing, shutil, platform, warnings
from builtins import print as _
tmp_bench_repo = '/tmp/.bench' tmp_bench_repo = '/tmp/.bench'
@ -9,7 +8,7 @@ log_path = os.path.join(tmp_log_folder, 'install_bench.log')
log_stream = sys.stdout log_stream = sys.stdout
def print(message, level=0): def log(message, level=0):
levels = { levels = {
0: '\033[94m', # normal 0: '\033[94m', # normal
1: '\033[92m', # success 1: '\033[92m', # success
@ -18,7 +17,7 @@ def print(message, level=0):
} }
start = levels.get(level) or '' start = levels.get(level) or ''
end = '\033[0m' end = '\033[0m'
_(start + message + end) print(start + message + end)
def find_executable(executable, path=None): def find_executable(executable, path=None):
@ -68,8 +67,8 @@ def check_environment():
message += "\nexport {}=C.UTF-8".format(var) message += "\nexport {}=C.UTF-8".format(var)
if message: if message:
print("Bench's CLI needs these to be defined!", level=3) log("Bench's CLI needs these to be defined!", level=3)
print("Run the following commands in shell: {}".format(message), level=2) log("Run the following commands in shell: {}".format(message), level=2)
sys.exit() sys.exit()
@ -99,15 +98,15 @@ def is_sudo_user():
def install_package(package): def install_package(package):
if find_executable(package): if find_executable(package):
print("{} already installed!".format(package), level=1) log("{} already installed!".format(package), level=1)
else: else:
print("Installing {}...".format(package)) log("Installing {}...".format(package))
success = run_os_command({ success = run_os_command({
'apt-get': ['sudo apt-get install -y {0}'.format(package)], 'apt-get': ['sudo apt-get install -y {0}'.format(package)],
'yum': ['sudo yum install -y {0}'.format(package)] 'yum': ['sudo yum install -y {0}'.format(package)]
}) })
if success: if success:
print("{} Installed!".format(package), level=1) log("{} Installed!".format(package), level=1)
else: else:
could_not_install(package) could_not_install(package)
@ -120,7 +119,7 @@ def install_bench(args):
if not os.path.exists(tmp_log_folder): if not os.path.exists(tmp_log_folder):
os.makedirs(tmp_log_folder) os.makedirs(tmp_log_folder)
log_stream = open(log_path, 'w') log_stream = open(log_path, 'w')
print("Logs are saved under {}".format(log_path), level=3) log("Logs are saved under {}".format(log_path), level=3)
check_distribution_compatibility() check_distribution_compatibility()
check_brew_installed() check_brew_installed()
@ -145,7 +144,7 @@ def install_bench(args):
}) })
if not success: if not success:
print('Could not install pre-requisites. Please check for errors or install them manually.') log('Could not install pre-requisites. Please check for errors or install them manually.')
return return
# secure pip installation # secure pip installation
@ -262,14 +261,14 @@ def check_distribution_compatibility():
'centos': [7] 'centos': [7]
} }
print("Checking System Compatibility...") log("Checking System Compatibility...")
if dist_name in supported_dists: if dist_name in supported_dists:
if float(dist_version) in supported_dists[dist_name]: if float(dist_version) in supported_dists[dist_name]:
print("{0} {1} is compatible!".format(dist_name, dist_version), level=1) log("{0} {1} is compatible!".format(dist_name, dist_version), level=1)
else: else:
print("Install on {0} {1} instead".format(dist_name, supported_dists[dist_name][-1]), level=3) log("Install on {0} {1} instead".format(dist_name, supported_dists[dist_name][-1]), level=3)
else: else:
print("Sorry, the installer doesn't support {0}. Aborting installation!".format(dist_name), level=2) log("Sorry, the installer doesn't support {0}. Aborting installation!".format(dist_name), level=2)
def get_distribution_info(): def get_distribution_info():
@ -374,7 +373,7 @@ def get_passwords(args):
with open(passwords_file_path, 'w') as f: with open(passwords_file_path, 'w') as f:
json.dump(passwords, f, indent=1) json.dump(passwords, f, indent=1)
print('Passwords saved at ~/passwords.txt') log('Passwords saved at ~/passwords.txt')
return passwords return passwords
@ -451,7 +450,7 @@ def parse_commandline_args():
if __name__ == '__main__': if __name__ == '__main__':
if not is_sudo_user(): if not is_sudo_user():
print("Please run this script as a non-root user with sudo privileges", level=3) log("Please run this script as a non-root user with sudo privileges", level=3)
sys.exit() sys.exit()
args = parse_commandline_args() args = parse_commandline_args()
@ -460,4 +459,4 @@ if __name__ == '__main__':
check_environment() check_environment()
install_bench(args) install_bench(args)
print("Bench + Frappe + ERPNext has been successfully installed!") log("Bench + Frappe + ERPNext has been successfully installed!")