unblock.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Simple unblock scriptlet"""
import sys
import requests


def unblock(ip):
    uid = None
    with open("blocklist.txt", "r") as f:
        for line in f:
            if line.startswith("# UUID: "):
                uid = line[8:].strip()
    if uid:
        print(f"Unblocking {ip}...")
        rv = requests.get(f"http://localhost:1729/aardvark-unblock?{ip}", headers={"X-Aardvark-Key": uid})
        print(rv.text)
    else:
        print("Could not locate Aardvark's UUID key! should be in blocklist.txt line 2 :/")


if __name__ == "__main__":
    if len(sys.argv) == 2:
        unblock(sys.argv[1])
    else:
        print("Usage: unblock.py [ip]")