utils.py 707 B

1234567891011121314151617181920212223242526272829303132333435
  1. from enum import Enum
  2. from urllib.parse import urlparse
  3. class Status(Enum):
  4. disconnect = 0
  5. connecting = 1
  6. connected = 2
  7. def parse_proxy_url(proxy_url):
  8. parsed_url = urlparse(proxy_url)
  9. scheme = parsed_url.scheme
  10. host = parsed_url.hostname
  11. port = parsed_url.port
  12. auth = None
  13. if parsed_url.username and parsed_url.password:
  14. auth = (parsed_url.username, parsed_url.password)
  15. return scheme, host, port, auth
  16. def parse_line(line):
  17. line = line.strip()
  18. if not line:
  19. return None, None
  20. if "==" in line:
  21. user_id, proxy_url = line.split('==')
  22. else:
  23. user_id, proxy_url = line, None
  24. return user_id, proxy_url or None