likho/microblog.py
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

added proxy settings for upload script

Browse Source
This commit is contained in:
likhy 2023-08-15 21:57:14 -07:00
parent 7d474af899
commit 26b7d47996
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
neouploader.py
View File

@ -1,11 +1,30 @@
import sys, subprocess, getpass, pycurl, urllib.parse import sys, os, subprocess, getpass, pycurl, urllib.parse
if __name__ == "__main__": if __name__ == "__main__":
def get_proxy():
proxy = ""
if "http_proxy" in os.environ:
proxy = os.environ["http_proxy"]
elif "https_proxy" in os.environ:
proxy = os.environ["https_proxy"]
host = proxy[proxy.rfind('/') + 1: proxy.rfind(':')]
port = proxy[proxy.rfind(':') + 1:]
foo = proxy.find("socks://") >= 0 or proxy.find("socks5h://")
return host, int(port), foo
def api_upload(endpoint, dest_fmt = "/microblog/%s"): def api_upload(endpoint, dest_fmt = "/microblog/%s"):
pages = [] pages = []
with open("updatedfiles.txt") as f: with open("updatedfiles.txt") as f:
pages = f.readlines() pages = f.readlines()
c = pycurl.Curl() c = pycurl.Curl()
if "http_proxy" in os.environ or "https_proxy" in os.environ:
hostname, port_no, is_socks = get_proxy()
c.setopt(pycurl.PROXY, hostname)
c.setopt(pycurl.PROXYPORT, port_no)
if is_socks:
c.setopt(pycurl.PROXYTYPE, pycurl.PROXYTYPE_SOCKS5_HOSTNAME)
c.setopt(c.URL, endpoint) c.setopt(c.URL, endpoint)
c.setopt(c.POST, 1) c.setopt(c.POST, 1)
for page in pages: for page in pages: