beter user detection by os module
This commit is contained in:
+3
-2
@@ -33,7 +33,7 @@ logging.basicConfig(
|
|||||||
ap_theme = Theme({'ok':'green', 'error':'red', 'checkt':'bold cyan','promp':'orange1'})
|
ap_theme = Theme({'ok':'green', 'error':'red', 'checkt':'bold cyan','promp':'orange1'})
|
||||||
console = Console(theme=ap_theme)
|
console = Console(theme=ap_theme)
|
||||||
|
|
||||||
user = input("Type user name: ")
|
user = os.getlogin()
|
||||||
home = os.path.join('/home',user)
|
home = os.path.join('/home',user)
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
@@ -47,7 +47,8 @@ def main():
|
|||||||
while(True):
|
while(True):
|
||||||
confirm_user = input(" y/n ")
|
confirm_user = input(" y/n ")
|
||||||
if (confirm_user.lower() == 'n'):
|
if (confirm_user.lower() == 'n'):
|
||||||
exit()
|
user = imput('Type username : ')
|
||||||
|
continue
|
||||||
elif (confirm_user.lower() == 'y'):
|
elif (confirm_user.lower() == 'y'):
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user