summaryrefslogtreecommitdiff
path: root/tasks.py
blob: 8cc5d98f6e4b2d5b79550ee7107358ad2878a890 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import urlparse

import requests


def purge_fastly_tags(domain, api_key, service_id, tags, max_tries=25):
    session = requests.session()
    headers = {"X-Fastly-Key": api_key, "Accept": "application/json"}

    all_tags = set(tags)
    purges = {}

    count = 0

    while all_tags and not count > max_tries:
        count += 1

        try:
            for tag in set(all_tags):
                # Build the URL
                url_path = "/service/%s/purge/%s" % (service_id, tag)
                url = urlparse.urljoin(domain, url_path)

                # Issue the Purge
                resp = session.post(url, headers=headers)
                resp.raise_for_status()

                # Store the Purge ID so we can track it later
                purges[tag] = resp.json()["id"]

            # for tag, purge_id in purges.iteritems():
            #     # Ensure that the purge completed successfully
            #     url = urlparse.urljoin(domain, "/purge")
            #     status = session.get(url, params={"id": purge_id})
            #     status.raise_for_status()

            #     # If the purge completely successfully remove the tag from
            #     #   our list.
            #     if status.json().get("results", {}).get("complete", None):
            #         all_tags.remove(tag)
        except Exception:
            if count > max_tries:
                raise