Advertisement
Guest User

Untitled

a guest
Mar 24th, 2016
173
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.08 KB | None | 0 0
  1. #!usr/bin/env python3
  2. import os
  3. import sys
  4. import base64
  5. import urllib.request
  6. import shutil
  7.  
  8. def checkEnv(name, strict=False):
  9. if not name in os.environ or os.environ[name] == '':
  10. if strict:
  11. print('Error need env var \'%s\'' % (name))
  12. sys.exit(22)
  13. return False
  14. return os.environ[name]
  15.  
  16. def download(url, dest, headers):
  17. req = urllib.request.Request(url)
  18. for header in headers:
  19. req.add_header(header, headers[header])
  20. with urllib.request.urlopen(req) as response, \
  21. open(dest, 'wb') as out_file:
  22. shutil.copyfileobj(response, out_file)
  23.  
  24. def main():
  25. username = 'AUTH_USER'
  26. username = checkEnv(username, strict=True)
  27. password = 'AUTH_PASS'
  28. password = checkEnv(password, strict=True)
  29. url = 'URL'
  30. url = checkEnv(url, strict=True)
  31. dest = 'DEST'
  32. dest = checkEnv(dest, strict=True)
  33.  
  34. concat = username + ":" + password
  35. concat = concat.encode('utf8')
  36. download(url, dest, {
  37. 'Autorization': 'Basic %s' % (base64.b64encode(concat))
  38. })
  39.  
  40. if __name__ == '__main__':
  41. main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement