@@ -22,7 +22,8 @@ def __init__(
2222 custom_domains : list = None ,
2323 crazy : bool = False ,
2424 ):
25- self .account_hash = self .hashed = account_hash
25+ self .account_hash = account_hash
26+ self .hashed = None
2627 self .separators = ["" , "." , "_" , "-" ]
2728 self .name_pattern = "[-_ ./]"
2829 self .emails = []
@@ -42,7 +43,7 @@ def __init__(
4243 self .n_combs = 0
4344 self .c = c (highlight = False )
4445
45- def load_domains (self ):
46+ def load_domains (self ) -> json :
4647 # Load domains from json files
4748 domain_files = {None : "" , "common" : "" , "long" : "_long" , "full" : "_full" }
4849 domain_file = domain_files [self .domain_list ]
@@ -65,13 +66,14 @@ def create_elements(self) -> list:
6566 )
6667 sys .exit (1 )
6768 # Get elements with custom arguments or from the Gravatar profile
69+ self .hashed = self .g .info ()["hash" ]
6870 if self .elements :
6971 elements = self .get_custom_elements ()
7072 else :
7173 elements = self .get_elements_from_gravatar ()
7274 return elements
7375
74- def get_custom_elements (self ):
76+ def get_custom_elements (self ) -> list :
7577 # Get elements from custom arguments
7678 self .elements = [
7779 unidecode (element .lower ())
@@ -85,19 +87,18 @@ def get_custom_elements(self):
8587 )
8688 return elements
8789
88- def get_elements_from_gravatar (self ):
90+ def get_elements_from_gravatar (self ) -> list :
8991 # Get elements from the Gravatar profile
9092 elements = []
9193 infos = self .g .info ()
9294 self .get_public_emails (infos )
93- self .hashed = infos ["hash" ]
9495 gob = self .process_gravatar_info (infos )
9596 for element in gob :
9697 deco = unidecode (element .lower ())
9798 elements .append (deco ) if deco not in elements else None
9899 return elements
99100
100- def get_public_emails (self , infos ) :
101+ def get_public_emails (self , infos : dict ) -> None :
101102 # Get emails from the Gravatar json emails
102103 if infos ["emails" ]:
103104 self .public_emails .extend (
@@ -110,7 +111,7 @@ def get_public_emails(self, infos):
110111 find = re .findall (pattern , infos ["aboutMe" ]) if infos ["aboutMe" ] else None
111112 self .public_emails .extend (find ) if find else None
112113
113- def process_gravatar_info (self , infos ) :
114+ def process_gravatar_info (self , infos : dict ) -> list :
114115 # Process Gravatar infos to get additional chunks
115116 gob = []
116117 self .add_preferred_username (infos , gob )
@@ -122,44 +123,44 @@ def process_gravatar_info(self, infos):
122123 self .add_elements (gob )
123124 return gob
124125
125- def add_preferred_username (self , infos , gob ) :
126+ def add_preferred_username (self , infos : dict , gob : list ) -> None :
126127 if infos ["preferredUsername" ]:
127128 gob .append (infos ["preferredUsername" ])
128129
129- def add_profile_url (self , infos , gob ) :
130+ def add_profile_url (self , infos : dict , gob : list ) -> None :
130131 if infos ["profileUrl" ]:
131132 gob .append (self .last_url_chunk (infos ["profileUrl" ]))
132133
133- def add_chunks (self , string , gob ) :
134+ def add_chunks (self , string : str , gob : list ) -> None :
134135 if string :
135136 names = re .split (self .name_pattern , unidecode (string ))
136137 chunks = [name for name in names if name ]
137138 # Add first letter for given and family names
138139 chunks .extend (name [:1 ] for name in names if len (name ) > 1 )
139140 gob .extend (chunks )
140141
141- def add_display_name (self , infos , gob ) :
142+ def add_display_name (self , infos : dict , gob : list ) -> None :
142143 names = re .split (self .name_pattern , unidecode (infos ["displayName" ]))
143144 gob .extend (name for name in names if name )
144145
145- def add_given_name (self , infos , gob ) :
146+ def add_given_name (self , infos : dict , gob : list ) -> None :
146147 # Add given name and first letter chunks
147148 if infos ["name" ] and infos ["name" ]["givenName" ]:
148149 self .add_chunks (infos ["name" ]["givenName" ], gob )
149150
150- def add_family_name (self , infos , gob ) :
151+ def add_family_name (self , infos : dict , gob : list ) -> None :
151152 # Add family name and first letter chunks
152153 if infos ["name" ] and infos ["name" ]["familyName" ]:
153154 self .add_chunks (infos ["name" ]["familyName" ], gob )
154155
155- def add_accounts (self , infos , gob ) :
156+ def add_accounts (self , infos : dict , gob : list ) -> None :
156157 # Add account chunks for verified accounts
157158 if infos ["accounts" ]:
158159 for account in infos ["accounts" ]:
159160 account_url = infos ["accounts" ][account ].rstrip ("/" )
160161 self .process_account (account , account_url , gob )
161162
162- def process_account (self , account , account_url , gob ) :
163+ def process_account (self , account : str , account_url : str , gob : list ) -> None :
163164 # Verified accounts username chunks
164165 if account in ["Mastodon" , "Fediverse" ]:
165166 gob .append (self .last_url_chunk (account_url ).replace ("@" , "" ))
@@ -200,7 +201,7 @@ def process_account(self, account, account_url, gob):
200201 chunk = self .last_url_chunk (account_url )
201202 gob .append (chunk )
202203
203- def add_elements (self , gob ) :
204+ def add_elements (self , gob : list ) -> None :
204205 # Building final list of chunks, deduped , lowercase and unidecoded if it hasn't been yet
205206 self .elements = []
206207 self .elements = [
@@ -209,27 +210,27 @@ def add_elements(self, gob):
209210 if unidecode (element .lower ()) not in self .elements
210211 ]
211212
212- def hash_email (self , email ) :
213+ def hash_email (self , email : str ) -> str :
213214 # MD5 hashing of a string email
214215 return hashlib .md5 (email .lower ().encode ()).hexdigest ()
215216
216- def check_mailhash (self , s : str ):
217+ def check_mailhash (self , s : str ) -> bool :
217218 # Check if a string is a valid MD5 hash
218219 return re .fullmatch (r"[a-fA-F0-9]{32}" , s ) is not None
219220
220- def check_email (self , email ) :
221+ def check_email (self , email : str ) -> bool :
221222 # Check if a string is a valid email
222223 return (
223224 True
224225 if re .match (r"(^[a-zA-Z0-9_.%+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)" , email )
225226 else False
226227 )
227228
228- def last_url_chunk (self , s : str ):
229+ def last_url_chunk (self , s : str ) -> str :
229230 # Get the last chunk of an URL
230231 return s .split ("/" )[- 1 ]
231232
232- def is_combination (self , s , chunks ) :
233+ def is_combination (self , s : str , chunks : list ) -> bool :
233234 # Logic to check if a string is a combination of other strings
234235 if s in chunks :
235236 chunks .remove (s )
@@ -244,18 +245,18 @@ def is_combination(self, s, chunks):
244245 return True
245246 return False
246247
247- def dedup_chunks (self , chunks ) :
248+ def dedup_chunks (self , chunks : list ) -> list :
248249 # Remove chunks from a list of chunks if it's made with other strings of the list
249250 return [s for s in chunks if not self .is_combination (s , chunks .copy ())]
250251
251- def show_chunks (self , elements ) :
252+ def show_chunks (self , elements : list ) -> str :
252253 # Show chunks as a string
253254 em = ""
254255 for element in elements :
255256 em += element + ", "
256257 return em .rstrip (", " )
257258
258- def get_combination_count (self , n ) :
259+ def get_combination_count (self , n : int ) -> int :
259260 # Calculate the total number of combinations for tdqm bar progress
260261 total = 0
261262 for r in range (1 , n + 1 ):
@@ -279,7 +280,7 @@ def get_combination_count(self, n):
279280 # Multiply by the number of domains
280281 return total * len (self .domains )
281282
282- def combinator (self ):
283+ def combinator (self ) -> str :
283284 # Generate all possible email combinations for unique elements
284285 # Get chunks and dedup them if made with other chunks
285286 elements = self .dedup_chunks (self .create_elements ())
@@ -323,7 +324,7 @@ def combinator(self):
323324 email_local_part = separator .join (permutation )
324325 yield f"{ email_local_part } { domain }
325326
326- def hashes (self ):
327+ def hashes (self ) -> str :
327328 # Calculate emails hash
328329 for email in self .combinator ():
329330 hashd = self .hash_email (email )
@@ -337,7 +338,7 @@ def hashes(self):
337338 return email
338339 self .bar .close ()
339340
340- def find (self ):
341+ def find (self ) -> None :
341342 # Print process and results
342343 self .c .print (
343344 f"[turquoise2]Finding email for [bold]{ self .account_hash } \n "
@@ -382,5 +383,8 @@ def find(self):
382383 "\n [light_pink1]:question_mark: Do you want to display the profile info? (y/n):[/light_pink1]"
383384 )
384385 if show_profile .lower () == "y" :
386+ print (
387+ f"------------------------------------------------------------------------------------\n "
388+ )
385389 self .g .print_info ()
386390 exit (0 )
0 commit comments