type checking enabled

Author: codeperfectplus

random_profile/__main__.py

@@ -1,4 +1,4 @@
-from .main import RandomProfile
+from audiobook.main import RandomProfile
 import argparse
 
 

random_profile/main.py

@@ -25,7 +25,6 @@
 cities_name_txt = os.path.join(ASSETS_DIR, "cities_name.txt")
 states_names_txt = os.path.join(ASSETS_DIR, "states_names.txt")
 job_titles_txt = os.path.join(ASSETS_DIR, "job_titles.txt")
-email_domain_txt = os.path.join(ASSETS_DIR, "email_domains.txt")
 
 # loading data from txt files
 fname = load_txt_file(fname_txt)
@@ -39,34 +38,34 @@
 
 
 class RandomProfile:
-    def __init__(self, num=1):
+    def __init__(self, num: int = 1):
         '''
         num = Total No. of Name You Want To Print
         default is 1
         To Print More Than one Name Change value of num
         '''
         self.num = num
 
-    def first_name(self, num=None):
+    def first_name(self, num: int = None) -> list:
         if num is None:
             num = self.num
         first_name_list = [random.choice(fname) for _ in range(num)]
         return first_name_list
 
-    def last_name(self, num=None):
+    def last_name(self, num: int = None) -> list:
         if num is None:
             num = self.num
         last_name_list = [random.choice(lname) for _ in range(num)]
         return last_name_list
 
-    def full_name(self, num=None):
+    def full_name(self, num: int = None) -> list:
         if num is None:
             num = self.num
         full_name_list = [random.choice(
             fname) + ' ' + random.choice(lname) for _ in range(num)]
         return full_name_list
 
-    def full_profile(self, num=None):
+    def full_profile(self, num: int = None) -> list:
         if num is None:
             num = self.num
         profile_list = []
@@ -113,10 +112,10 @@ def full_profile(self, num=None):
 
         return profile_list
 
-    def ipv4(self):
+    def ipv4(self) -> list:
         ip_list = [ipv4_gen() for _ in range(self.num)]
         return ip_list
 
-    def job_title(self):
+    def job_title(self) -> list:
         job_title_list = [random.choice(job_titles) for _ in range(self.num)]
         return job_title_list

random_profile/utils.py

@@ -28,7 +28,7 @@ def ipv4_gen() -> str:
     return f"{random.randint(0,255)}.{random.randint(0,255)}.{random.randint(0,255)}.{random.randint(0,255)}"
 
 
-def generate_dob_age():
+def generate_dob_age() -> tuple:
     month = random.randint(1, 12)
     if month == 2:  # if month is feb
         day = random.randint(1, 28)
@@ -44,10 +44,10 @@ def generate_dob_age():
     age = (datetime.now() - dob).days // 365
     dob = dob.strftime("%d/%m/%Y")
 
-    return dob, age
+    return (dob, age)
 
 
-def generate_random_height_weight():
+def generate_random_height_weight() -> tuple:
     height = random.randint(140, 200)
     if height < 150:
         weight = random.randint(40, 60)
@@ -61,4 +61,4 @@ def generate_random_height_weight():
         weight = random.randint(80, 100)
     elif height <= 200:
         weight = random.randint(90, 110)
-    return height, weight
+    return (height, weight)