File: example_firewall_rules.py

package info (click to toggle)
python-cloudflare 2.20.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,056 kB
  • sloc: python: 6,932; makefile: 138; sh: 76
file content (116 lines) | stat: -rw-r--r-- 3,512 bytes parent folder | download | duplicates (2)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#!/usr/bin/env python
"""Cloudflare API code - example"""

import os
import sys
import re
import json
import uuid

sys.path.insert(0, os.path.abspath('.'))
sys.path.insert(0, os.path.abspath('..'))
import CloudFlare

def main():
    """Cloudflare API code - example"""

    cf = CloudFlare.CloudFlare()

    try:
        zone_name = sys.argv[1]
    except IndexError:
        exit('usage: example_firewall_rules.py zone_name')

    # grab the zone identifier
    try:
        params = {'name': zone_name}
        zones = cf.zones.get(params=params)
    except CloudFlare.exceptions.CloudFlareAPIError as e:
        exit('/zone %d %s - api call failed' % (e, e))
    except Exception as e:
        exit('/zone.get - %s - api call failed' % (e))

    if len(zones) == 0:
        exit('/zones.get - %s - zones not found' % (zone_name))

    if len(zones) != 1:
        exit('/zones.get - %s - api call returned %d items' % (zone_name, len(zones)))

    zone_id = zones[0]['id']

    # SHOW EXISTING FIREWALL RULES
    r = cf.zones.firewall.rules.get(zone_id)
    print('existing filewall rules =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    # SHOW EXISTING FILTERS
    r = cf.zones.filters.get(zone_id)
    print('existing filters =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    # CREATE A FILTER & FIREWALL RULES

    reference_name = 'FILTER-' + str(uuid.uuid1())

    my_filter = {
        'expression': 'http.request.uri.path == "/private.html$"',
        'paused': True,
        'description': 'stop access to /private.html',
        'ref': reference_name,
    }

    my_rule = [
        {
            'action': 'block',
            'filter': my_filter,
            'paused': True,
        }
    ]

    try:
        r = cf.zones.firewall.rules.post(zone_id, data=my_rule)
    except CloudFlare.exceptions.CloudFlareAPIError as e:
        print('create zones.filewall.rules: %d %s' % (int(e), str(e)))
        exit(1)

    print('firewall rule created =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    firewall_id = r[0]['id']
    filter_id = r[0]['filter']['id']

    print('filewall_id = %s filter_id = %s' % (firewall_id, filter_id))

    # SHOW PRESENT FIREWALL RULES
    r = cf.zones.firewall.rules.get(zone_id)
    print('present filewall rules =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    # DELETE NEW FIREWALL RULES
    for f in r:
        print('id = ' + f['id'])
        try:
            r2 = cf.zones.firewall.rules.delete(zone_id, f['id'])
            print('deleted id = ' + r2['id'])
        except CloudFlare.exceptions.CloudFlareAPIError as e:
            print('zones.filewall.rules.delete: %d %s' % (int(e), str(e)))

    # SHOW PRESENT FILTERS
    r = cf.zones.filters.get(zone_id)
    print('present filters =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    # DELETE NEW FILTERS
    for f in r:
        print('id = ' + f['id'])
        try:
            r2 = cf.zones.filters.delete(zone_id, f['id'])
            print('deleted id = ' + r2['id'])
        except CloudFlare.exceptions.CloudFlareAPIError as e:
            print('zones.filters.delete: %d %s' % (int(e), str(e)))

    # SHOW FINAL FIREWALL RULES
    r = cf.zones.firewall.rules.get(zone_id)
    print('final filewall rules =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

    # SHOW FINAL FILTERS
    r = cf.zones.filters.get(zone_id)
    print('final filters =\n' + json.dumps(r, indent=4, sort_keys=False) + '\n')

if __name__ == '__main__':
    main()