twitools/httptools/__init__.py

23 lines
799 B
Python
Raw Normal View History

import codecs, setuptools, pyshorteners, requests, twitools, sys
def downloadMedia(url, tid, mid, two = twitools.twoHelper()):
remote = two.auth.oauth.get(url, auth=two.api.auth.apply_auth())
filename = "media/%s_%i.%s" % (str(tid), int(mid), url.split(".")[-1])
with open(filename, 'wb') as outfile:
for chunk in remote.iter_content(chunk_size=1024):
if chunk:
outfile.write(chunk)
outfile.flush()
2017-03-13 21:55:49 +00:00
def longURL(url):
2017-03-13 22:31:05 +00:00
return requests.get(url, allow_redirects=True).url
def shortURL(url):
try:
2017-03-13 21:55:49 +00:00
return pyshorteners.Shortener('Google', api_key=setuptools.getSetting("Google", "googl")).short(longURL(url))
except pyshorteners.exceptions.ShorteningErrorException as e:
codecs.getwriter("utf8")(sys.stdout.buffer).write("Error handling %s\n" % url)
2017-03-13 21:55:49 +00:00
print(e)
return False