From 45fc1c6d55e175562d933ac17148b5ac31262020 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mateusz=20Gruszczy=C5=84ski?= Date: Thu, 5 Mar 2026 08:19:37 +0100 Subject: [PATCH] add NI --- app.py | 3 ++- config.py | 4 +++- script.py | 13 ++++++------- 3 files changed, 11 insertions(+), 9 deletions(-) diff --git a/app.py b/app.py index ed367fc..2ca7e86 100644 --- a/app.py +++ b/app.py @@ -4,6 +4,7 @@ GeoIP Ban Generator - Web Application from flask import Flask, render_template, request, Response, jsonify from geoip_handler import GeoIPHandler +from flask import jsonify, render_template, request from pathlib import Path from functools import wraps from datetime import datetime @@ -116,7 +117,7 @@ def inject_globals(): 'redis_connected': redis_cache.health_check()['connected'] if redis_cache else False, } -from flask import jsonify, render_template, request + def _wants_json(): if request.path.startswith("/api/"): diff --git a/config.py b/config.py index 31aaf26..075279f 100644 --- a/config.py +++ b/config.py @@ -189,6 +189,8 @@ COMMON_COUNTRIES = [ {'code': 'EC', 'name': 'Ecuador'}, {'code': 'BO', 'name': 'Bolivia'}, {'code': 'PY', 'name': 'Paraguay'}, + {'code': 'NI', 'name': 'Nicaragua'}, + # Caribbean {'code': 'CU', 'name': 'Cuba'}, @@ -306,4 +308,4 @@ MAXMIND_CHUNK_MIN = int(os.getenv('MAXMIND_CHUNK_MIN', '200')) MAXMIND_CHUNK_MAX = int(os.getenv('MAXMIND_CHUNK_MAX', '4000')) MAXMIND_WORKERS_MIN = int(os.getenv('MAXMIND_WORKERS_MIN', '6')) -MAXMIND_WORKERS_MAX = int(os.getenv('MAXMIND_WORKERS_MAX', '48')) \ No newline at end of file +MAXMIND_WORKERS_MAX = int(os.getenv('MAXMIND_WORKERS_MAX', '48')) diff --git a/script.py b/script.py index eacea39..012593e 100644 --- a/script.py +++ b/script.py @@ -4,17 +4,18 @@ import re import requests BASE_URL = "http://geo-block.krk.itg.demo-ht.iadm" -ENDPOINT = "/api/generate" +ENDPOINT = "/api/generate/raw" -payload = { +payload_json = """{ "countries": [ "PL" ], "aggregate": true, "use_cache": true, - "app_type": "haproxy", - "app_variant": "map" -} + "app_type": "raw-cidr_json", + "as_js": false +}""" +payload = json.loads(payload_json) resp = requests.post(BASE_URL + ENDPOINT, json=payload, timeout=120) @@ -26,7 +27,6 @@ print("X-Generated-At:", resp.headers.get("X-Generated-At")) ct = (resp.headers.get("Content-Type") or "").lower() if resp.status_code >= 400: - # try show JSON error, else text try: print(json.dumps(resp.json(), indent=2)) except Exception: @@ -42,7 +42,6 @@ else: if m: filename = m.group(1) else: - # fallback extension if "text/csv" in ct: filename += ".csv" elif "javascript" in ct: