Added features to the password generator

New support for digits and special characters

Author: root

Password Generator/pass_gen.py

@@ -1,13 +1,24 @@
-'''
-The secrets module is used for generating cryptographically strong random numbers suitable for managing data such as passwords,
-account authentication, security tokens, and related secrets.
-'''
-import string
+import string as str
 import secrets
+class PasswordGenerator():
 
-def secure_password_gen(passlength):
-    password = ''.join((secrets.choice(string.ascii_letters) for i in range(passlength)))
-    return password
+    @staticmethod
+    def set_sequence(*conditions): #must have 4 conditions, one for each menber of the list possible_characters
+        possible_characters=[str.ascii_lowercase, str.ascii_uppercase, str.digits, str.punctuation]
+        sequence=""
+        for x in range(len(conditions)):
+            if conditions[x]:
+                sequence+=possible_characters[x]
+            else:
+                pass
+        return sequence
 
-n = int(input('Enter length of password : '))
-print('Password generated is :', secure_password_gen(n))
+
+    @staticmethod
+    def secure_password_gen(passlength=8, ):
+        characters_for_password = PasswordGenerator.set_sequence(True, True, True, True)
+        password = ''.join((secrets.choice(characters_for_password) for i in range(passlength)))
+        return password
+
+while True:
+    print('Password generated is :', PasswordGenerator.secure_password_gen(int(input("Enter password lenght: "))))