Implement bashrc module code
This commit is contained in:
parent
b1a34aabd4
commit
1d4e6a3f8d
4
main.py
4
main.py
@ -8,6 +8,7 @@ from modules.vim.VimModule import VimModule
|
||||
from modules.teamspeak.Teamspeak import Teamspeak
|
||||
from modules.hacker.Hacker import Hacker
|
||||
import modules.fail2ban.Fail2BanModule as fail2ban
|
||||
from modules.bashrc.BashrcModule import BashrcModule
|
||||
|
||||
modules = {
|
||||
"helloworld": HelloWorld().run,
|
||||
@ -17,7 +18,8 @@ modules = {
|
||||
"vim": VimModule().run,
|
||||
"teamspeak": Teamspeak,
|
||||
"hacker": Hacker,
|
||||
"fail2ban": fail2ban.run()
|
||||
"fail2ban": fail2ban.run(),
|
||||
"bashrc": BashrcModule().run
|
||||
}
|
||||
|
||||
fire.Fire(modules)
|
||||
|
26
modules/bashrc/BashrcModule.py
Normal file
26
modules/bashrc/BashrcModule.py
Normal file
@ -0,0 +1,26 @@
|
||||
from PyInquirer import prompt
|
||||
import os
|
||||
from shutil import copyfile
|
||||
from print_helpers import print_gr, print_warn, print_fail
|
||||
|
||||
|
||||
class BashrcModule:
|
||||
|
||||
def run(self):
|
||||
bashrc_path = os.path.join(os.path.expanduser('~'), ".bashrc")
|
||||
bashrc_exists = os.path.exists(bashrc_path)
|
||||
|
||||
if bashrc_exists:
|
||||
print_gr(f"Found .bashrc in {bashrc_path}")
|
||||
ask_to_keep = {
|
||||
'type': 'confirm',
|
||||
'message': 'Do you want to overwrite the existing .bashrc?',
|
||||
'name': 'overwrite',
|
||||
'default': False,
|
||||
}
|
||||
if not prompt(ask_to_keep)['overwrite']:
|
||||
print_fail("Stopping")
|
||||
return
|
||||
|
||||
copyfile("./modules/bashrc/bashrc_default", bashrc_path)
|
||||
print_gr("Overwritten") if bashrc_exists else print_gr("Created")
|
Loading…
Reference in New Issue
Block a user