bintray.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import json
  4. import requests
  5. from .const import NAME
  6. class BintrayAPI(requests.Session):
  7. def __init__(self, api_key, user, *args, **kwargs):
  8. super(BintrayAPI, self).__init__(*args, **kwargs)
  9. self.auth = (user, api_key)
  10. self.base_url = 'https://api.bintray.com/'
  11. def create_repository(self, subject, repo_name, repo_type='generic'):
  12. url = '{base}repos/{subject}/{repo_name}'.format(
  13. base=self.base_url, subject=subject, repo_name=repo_name,
  14. )
  15. data = {
  16. 'name': repo_name,
  17. 'type': repo_type,
  18. 'private': False,
  19. 'desc': 'Automated release for {}: {}'.format(NAME, repo_name),
  20. 'labels': ['docker-compose', 'docker', 'release-bot'],
  21. }
  22. return self.post_json(url, data)
  23. def repository_exists(self, subject, repo_name):
  24. url = '{base}/repos/{subject}/{repo_name}'.format(
  25. base=self.base_url, subject=subject, repo_name=repo_name,
  26. )
  27. result = self.get(url)
  28. if result.status_code == 404:
  29. return False
  30. result.raise_for_status()
  31. return True
  32. def delete_repository(self, subject, repo_name):
  33. url = '{base}repos/{subject}/{repo_name}'.format(
  34. base=self.base_url, subject=subject, repo_name=repo_name,
  35. )
  36. return self.delete(url)
  37. def post_json(self, url, data, **kwargs):
  38. if 'headers' not in kwargs:
  39. kwargs['headers'] = {}
  40. kwargs['headers']['Content-Type'] = 'application/json'
  41. return self.post(url, data=json.dumps(data), **kwargs)