quantum/purge-static
1
0
Fork 0
mirror of https://github.com/quantum5/purge-static.git synced 2025-04-24 11:22:00 -04:00
Code Issues Packages Projects Releases Wiki Activity

Implement Cloudflare API token support

Browse source
This commit is contained in:
Quantum 2022-01-27 22:19:49 -05:00
parent 47dc38f4ba
commit c31c04eaa6
1 changed files with 22 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
purge_static/cdn/cloudflare.py
View file

@ -22,26 +22,38 @@ class CloudFlareCDN(object):
except ValueError: except ValueError:
sys.exit('Credentials file not valid JSON: %s' % (args.credentials,)) sys.exit('Credentials file not valid JSON: %s' % (args.credentials,))
self.email = credentials.get('email') self.api_token = credentials.get('api_token')
if not isinstance(self.email, six.string_types): if self.api_token:
sys.exit('In credentials file: key "email" should map to a string') if not isinstance(self.api_token, six.string_types):
sys.exit('In credentials file: key "api_token" should map to a string')
self.api_key = credentials.get('api_key') self.email = self.api_key = None
if not isinstance(self.api_key, six.string_types): else:
sys.exit('In credentials file: key "api_key" should map to a string') self.email = credentials.get('email')
if not isinstance(self.email, six.string_types):
sys.exit('In credentials file: key "email" should map to a string')
self.api_key = credentials.get('api_key')
if not isinstance(self.api_key, six.string_types):
sys.exit('In credentials file: key "api_key" should map to a string')
self.zone = args.zone self.zone = args.zone
if not self.zone: if not self.zone:
sys.exit('No zone for CloudFlare, use --zone.') sys.exit('No zone for CloudFlare, use --zone.')
def purge(self, urls): def purge(self, urls):
if self.api_token:
headers = {'Authorization': f'Bearer {self.api_token}'}
else:
headers = {
'X-Auth-Email': self.email,
'X-Auth-Key': self.api_key,
}
for group in chunk(urls, CLOUDFLARE_MAX_PURGE): for group in chunk(urls, CLOUDFLARE_MAX_PURGE):
resp = requests.post( resp = requests.post(
'https://api.cloudflare.com/client/v4/zones/%s/purge_cache' % (self.zone,), 'https://api.cloudflare.com/client/v4/zones/%s/purge_cache' % (self.zone,),
json={'files': group}, headers={ json={'files': group}, headers=headers
'X-Auth-Email': self.email,
'X-Auth-Key': self.api_key,
}
).json() ).json()
if not resp.get('success'): if not resp.get('success'):