mirror of https://github.com/dnomd343/ProxyC
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
578 B
21 lines
578 B
2 years ago
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from IPy import IP
|
||
|
|
|
||
|
|
|
||
|
|
def v6AddBracket(host: str) -> str: # add bracket for ipv6
|
||
|
|
return hostFormat(host, v6Bracket = True)
|
||
|
|
|
||
|
|
|
||
|
|
def hostFormat(host: str, v6Bracket: bool = False) -> str:
|
||
|
|
try:
|
||
|
|
if host[:1] == '[' and host[-1:] == ']': # [IPv6]
|
||
|
|
host = host[1:-1] # remove extra bracket
|
||
|
|
ip = IP(host)
|
||
|
|
if v6Bracket and ip.version() == 6:
|
||
|
|
return '[%s]' % str(ip) # [IPv6]
|
||
|
|
return str(ip) # IPv4 / IPV6
|
||
|
|
except: # not ip address
|
||
|
|
return host
|