Fix subdomain
This commit is contained in:
@@ -29,7 +29,7 @@ def GetDomainlevel(domain):
|
|||||||
"""
|
"""
|
||||||
This function returns the domain type : domain, subdomain,
|
This function returns the domain type : domain, subdomain,
|
||||||
"""
|
"""
|
||||||
domain_name = domain.lower().split('.')
|
domain_name = domain.lower().strip().split('.')
|
||||||
if domain_name[0] == 'www':
|
if domain_name[0] == 'www':
|
||||||
domain_name = domain_name[1:]
|
domain_name = domain_name[1:]
|
||||||
domain_type = ''
|
domain_type = ''
|
||||||
@@ -39,16 +39,17 @@ def GetDomainlevel(domain):
|
|||||||
"/var/lib/wo/public_suffix_list.dat", encoding='utf-8', )
|
"/var/lib/wo/public_suffix_list.dat", encoding='utf-8', )
|
||||||
# Read all the lines into a list.
|
# Read all the lines into a list.
|
||||||
for domain_suffix in Suffix_file:
|
for domain_suffix in Suffix_file:
|
||||||
if (str(domain_suffix).strip()) == ('.'.join(domain_name[1:])):
|
if (str(domain_suffix).strip()) == ('.'.join(domain_name[2:])):
|
||||||
domain_type = 'domain'
|
|
||||||
root_domain = ('.'.join(domain_name[0:]))
|
|
||||||
break
|
|
||||||
elif (str(domain_suffix).strip()) == ('.'.join(domain_name[2:])):
|
|
||||||
domain_type = 'subdomain'
|
domain_type = 'subdomain'
|
||||||
root_domain = ('.'.join(domain_name[1:]))
|
root_domain = ('.'.join(domain_name[1:]))
|
||||||
break
|
break
|
||||||
|
elif (str(domain_suffix).strip()) == ('.'.join(domain_name[1:])):
|
||||||
|
domain_type = 'domain'
|
||||||
|
root_domain = ('.'.join(domain_name))
|
||||||
|
break
|
||||||
else:
|
else:
|
||||||
domain_type = 'other'
|
domain_type = 'domain'
|
||||||
|
root_domain = ('.'.join(domain_name))
|
||||||
Suffix_file.close()
|
Suffix_file.close()
|
||||||
|
|
||||||
return (domain_type, root_domain)
|
return (domain_type, root_domain)
|
||||||
|
|||||||
Reference in New Issue
Block a user