Compare commits
8 Commits
4eb906f6d4
...
handle_cor
| Author | SHA1 | Date | |
|---|---|---|---|
|
8ec34cfbeb
|
|||
|
479f990044
|
|||
|
078514b69e
|
|||
|
7754b77b82
|
|||
|
87a9b1dd5f
|
|||
|
98643f9a69
|
|||
|
e77e4dc174
|
|||
| a95dfaae2b |
17
.gitea/workflows/bandit.yml
Normal file
17
.gitea/workflows/bandit.yml
Normal file
@ -0,0 +1,17 @@
|
||||
---
|
||||
name: Bandit
|
||||
on: [push]
|
||||
|
||||
|
||||
# XXX need to do stuff with uv
|
||||
jobs:
|
||||
build:
|
||||
runs-on: freebsd
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ["3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Analyse code with Bandit
|
||||
run: |
|
||||
bandit -r .
|
||||
17
.gitea/workflows/flake8.yml
Normal file
17
.gitea/workflows/flake8.yml
Normal file
@ -0,0 +1,17 @@
|
||||
---
|
||||
name: Flake8
|
||||
on: [push]
|
||||
|
||||
|
||||
# XXX need to do stuff with uv
|
||||
jobs:
|
||||
build:
|
||||
runs-on: freebsd
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ["3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Analyse code with Flake8
|
||||
run: |
|
||||
flake8 $(git ls-files '*.py')
|
||||
17
.gitea/workflows/mypy.yml
Normal file
17
.gitea/workflows/mypy.yml
Normal file
@ -0,0 +1,17 @@
|
||||
---
|
||||
name: Mypy
|
||||
on: [push]
|
||||
|
||||
|
||||
# XXX need to do stuff with uv
|
||||
jobs:
|
||||
build:
|
||||
runs-on: freebsd
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ["3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Analyse code with Mypy
|
||||
run: |
|
||||
mypy --install-types --non-interactive $(git ls-files '*.py')
|
||||
23
.gitea/workflows/pip-audit.yml
Normal file
23
.gitea/workflows/pip-audit.yml
Normal file
@ -0,0 +1,23 @@
|
||||
---
|
||||
name: pip-audit
|
||||
on:
|
||||
push:
|
||||
branches: [main]
|
||||
pull_request:
|
||||
branches: [main]
|
||||
schedule:
|
||||
- cron: '0 0 * * 0' # Weekly on Sunday
|
||||
|
||||
# XXX need to do stuff with uv
|
||||
jobs:
|
||||
build:
|
||||
runs-on: freebsd
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ["3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Check vulnerable components with pip-audit
|
||||
run: |
|
||||
pip-audit .
|
||||
|
||||
17
.gitea/workflows/pylint.yml
Normal file
17
.gitea/workflows/pylint.yml
Normal file
@ -0,0 +1,17 @@
|
||||
---
|
||||
name: Pylint
|
||||
on: [push]
|
||||
|
||||
|
||||
# XXX need to do stuff with uv
|
||||
jobs:
|
||||
build:
|
||||
runs-on: freebsd
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ["3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Analyse code with Pylint
|
||||
run: |
|
||||
pylint $(git ls-files '*.py')
|
||||
2
.gitignore
vendored
2
.gitignore
vendored
@ -160,3 +160,5 @@ cython_debug/
|
||||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
||||
|
||||
# Vim swap files
|
||||
*.sw?
|
||||
|
||||
29
app/main.py
29
app/main.py
@ -1,19 +1,40 @@
|
||||
'''
|
||||
Simple Geolocation with FastAPI
|
||||
'''
|
||||
import os
|
||||
from ipaddress import IPv4Address, IPv4Network, IPv6Address, IPv6Network
|
||||
from typing import Annotated, Optional, Union
|
||||
|
||||
import geoip2.database
|
||||
from geoip2.errors import AddressNotFoundError
|
||||
from fastapi import FastAPI, Path, Body, Request, Response, status
|
||||
from dotenv import load_dotenv
|
||||
from fastapi import Body, FastAPI, Path, Request, Response, status
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.responses import RedirectResponse
|
||||
from geoip2.errors import AddressNotFoundError
|
||||
from pydantic import BaseModel
|
||||
|
||||
# Load environment variables
|
||||
load_dotenv()
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
GEOLITE2_ASN_DB = '/usr/local/share/GeoIP/GeoLite2-ASN.mmdb'
|
||||
GEOLITE2_CITY_DB = '/usr/local/share/GeoIP/GeoLite2-City.mmdb'
|
||||
# Configure CORS from environment variables
|
||||
cors_origins = os.getenv('CORS_ALLOW_ORIGINS', 'http://localhost')
|
||||
allow_origins = [origin.strip() for origin in cors_origins.split(',')
|
||||
if origin.strip()]
|
||||
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=allow_origins,
|
||||
allow_credentials=True,
|
||||
allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
GEOLITE2_ASN_DB = os.getenv('GEOLITE2_ASN_DB',
|
||||
'/usr/local/share/GeoIP/GeoLite2-ASN.mmdb')
|
||||
GEOLITE2_CITY_DB = os.getenv('GEOLITE2_CITY_DB',
|
||||
'/usr/local/share/GeoIP/GeoLite2-City.mmdb')
|
||||
|
||||
|
||||
class IPAddressParam(BaseModel):
|
||||
|
||||
2
requirements.txt
Normal file
2
requirements.txt
Normal file
@ -0,0 +1,2 @@
|
||||
geoip2==4.7.0
|
||||
fastapi==0.115.6
|
||||
Reference in New Issue
Block a user