484 lines
14 KiB
Python
484 lines
14 KiB
Python
|
from flask import Flask, request, jsonify, render_template_string
|
||
|
|
from datetime import date
|
||
|
|
import re
|
||
|
|
import requests
|
||
|
|
from requests import Session
|
||
|
|
from zeep import Client
|
||
|
|
from zeep.transports import Transport
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
VIES_WSDL_URL = "https://ec.europa.eu/taxation_customs/vies/services/checkVatService.wsdl"
|
||
|
|
POLISH_WL_API = "https://wl-api.mf.gov.pl/api/search/nip/{nip}?date={query_date}"
|
||
|
|
|
||
|
|
|
||
|
|
def get_vies_client():
|
||
|
|
session = Session()
|
||
|
|
transport = Transport(session=session, timeout=20)
|
||
|
|
return Client(wsdl=VIES_WSDL_URL, transport=transport)
|
||
|
|
|
||
|
|
|
||
|
|
def normalize_vat(country_code: str, vat_number: str):
|
||
|
|
country_code = (country_code or "").strip().upper()
|
||
|
|
vat_number = (vat_number or "").strip().replace(" ", "")
|
||
|
|
if vat_number.upper().startswith(country_code):
|
||
|
|
vat_number = vat_number[len(country_code):]
|
||
|
|
return country_code, vat_number
|
||
|
|
|
||
|
|
|
||
|
|
def split_zip_city(text: str):
|
||
|
|
if not text:
|
||
|
|
return "", ""
|
||
|
|
text = " ".join(text.split())
|
||
|
|
m = re.match(r"^(\d{2}-\d{3})\s+(.+)$", text)
|
||
|
|
if m:
|
||
|
|
return m.group(1), m.group(2)
|
||
|
|
return "", text
|
||
|
|
|
||
|
|
|
||
|
|
def parse_polish_working_address(address: str):
|
||
|
|
if not address:
|
||
|
|
return "", "", ""
|
||
|
|
address = " ".join(address.split())
|
||
|
|
m = re.match(r"^(.*?)(\d{2}-\d{3})\s+(.+)$", address)
|
||
|
|
if m:
|
||
|
|
street = m.group(1).strip(" ,")
|
||
|
|
zip_code = m.group(2).strip()
|
||
|
|
city = m.group(3).strip()
|
||
|
|
return street, zip_code, city
|
||
|
|
return address, "", ""
|
||
|
|
|
||
|
|
|
||
|
|
def parse_vies_address(address: str):
|
||
|
|
if not address:
|
||
|
|
return "", "", ""
|
||
|
|
|
||
|
|
parts = [p.strip() for p in address.split(",") if p.strip()]
|
||
|
|
if not parts:
|
||
|
|
return "", "", ""
|
||
|
|
|
||
|
|
street = parts[0]
|
||
|
|
last_part = parts[-1]
|
||
|
|
|
||
|
|
# Common cases:
|
||
|
|
# "1000 Brussels"
|
||
|
|
# "00-838 Warszawa"
|
||
|
|
# "HR-10000 Zagreb"
|
||
|
|
m = re.match(r"^([A-Z]{0,2}-?\d[\dA-Z -]*)\s+(.+)$", last_part)
|
||
|
|
if m:
|
||
|
|
zip_code = m.group(1).strip()
|
||
|
|
city = m.group(2).strip()
|
||
|
|
return street, zip_code, city
|
||
|
|
|
||
|
|
return street, "", last_part
|
||
|
|
|
||
|
|
|
||
|
|
def check_vies(country_code: str, vat_number: str):
|
||
|
|
client = get_vies_client()
|
||
|
|
result = client.service.checkVat(countryCode=country_code, vatNumber=vat_number)
|
||
|
|
|
||
|
|
name = (result.name or "").strip()
|
||
|
|
address = (result.address or "").strip()
|
||
|
|
street, zip_code, city = parse_vies_address(address)
|
||
|
|
|
||
|
|
return {
|
||
|
|
"source": "vies",
|
||
|
|
"found": bool(result.valid),
|
||
|
|
"country": country_code,
|
||
|
|
"vat_number": vat_number,
|
||
|
|
"name": name if result.valid else "",
|
||
|
|
"street": street if result.valid else "",
|
||
|
|
"zip": zip_code if result.valid else "",
|
||
|
|
"city": city if result.valid else "",
|
||
|
|
"raw_address": address if result.valid else "",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def check_polish_wl(nip: str):
|
||
|
|
today = date.today().isoformat()
|
||
|
|
url = POLISH_WL_API.format(nip=nip, query_date=today)
|
||
|
|
|
||
|
|
response = requests.get(url, headers={"Accept": "application/json"}, timeout=20)
|
||
|
|
response.raise_for_status()
|
||
|
|
|
||
|
|
data = response.json()
|
||
|
|
result = data.get("result", {})
|
||
|
|
subject = result.get("subject")
|
||
|
|
|
||
|
|
if not subject:
|
||
|
|
subjects = result.get("subjects") or []
|
||
|
|
subject = subjects[0] if subjects else None
|
||
|
|
|
||
|
|
if not subject:
|
||
|
|
return {
|
||
|
|
"source": "polish_wl",
|
||
|
|
"found": False,
|
||
|
|
"country": "PL",
|
||
|
|
"vat_number": nip,
|
||
|
|
"name": "",
|
||
|
|
"street": "",
|
||
|
|
"zip": "",
|
||
|
|
"city": "",
|
||
|
|
"raw_address": "",
|
||
|
|
}
|
||
|
|
|
||
|
|
name = (subject.get("name") or "").strip()
|
||
|
|
working_address = (subject.get("workingAddress") or "").strip()
|
||
|
|
street, zip_code, city = parse_polish_working_address(working_address)
|
||
|
|
|
||
|
|
return {
|
||
|
|
"source": "polish_wl",
|
||
|
|
"found": True,
|
||
|
|
"country": "PL",
|
||
|
|
"vat_number": nip,
|
||
|
|
"name": name,
|
||
|
|
"street": street,
|
||
|
|
"zip": zip_code,
|
||
|
|
"city": city,
|
||
|
|
"raw_address": working_address,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/api/company_lookup")
|
||
|
|
def company_lookup():
|
||
|
|
country = request.args.get("country", "")
|
||
|
|
vat_number = request.args.get("vat_number", "")
|
||
|
|
|
||
|
|
country, vat_number = normalize_vat(country, vat_number)
|
||
|
|
|
||
|
|
if not country or not vat_number:
|
||
|
|
return jsonify({"error": "country and vat_number are required"}), 400
|
||
|
|
|
||
|
|
try:
|
||
|
|
vies_result = check_vies(country, vat_number)
|
||
|
|
except Exception as e:
|
||
|
|
vies_result = {
|
||
|
|
"source": "vies",
|
||
|
|
"found": False,
|
||
|
|
"country": country,
|
||
|
|
"vat_number": vat_number,
|
||
|
|
"name": "",
|
||
|
|
"street": "",
|
||
|
|
"zip": "",
|
||
|
|
"city": "",
|
||
|
|
"raw_address": "",
|
||
|
|
"warning": f"VIES lookup failed: {e}",
|
||
|
|
}
|
||
|
|
|
||
|
|
if vies_result.get("found"):
|
||
|
|
return jsonify(vies_result)
|
||
|
|
|
||
|
|
if country == "PL":
|
||
|
|
try:
|
||
|
|
polish_result = check_polish_wl(vat_number)
|
||
|
|
if "warning" in vies_result:
|
||
|
|
polish_result["warning"] = vies_result["warning"]
|
||
|
|
return jsonify(polish_result)
|
||
|
|
except Exception as e:
|
||
|
|
response = {
|
||
|
|
"source": "polish_wl",
|
||
|
|
"found": False,
|
||
|
|
"country": country,
|
||
|
|
"vat_number": vat_number,
|
||
|
|
"name": "",
|
||
|
|
"street": "",
|
||
|
|
"zip": "",
|
||
|
|
"city": "",
|
||
|
|
"raw_address": "",
|
||
|
|
"warning": vies_result.get("warning"),
|
||
|
|
"fallback_error": f"Polish WL lookup failed: {e}",
|
||
|
|
}
|
||
|
|
return jsonify(response)
|
||
|
|
|
||
|
|
return jsonify(vies_result)
|
||
|
|
|
||
|
|
|
||
|
|
HTML = """
|
||
|
|
<!DOCTYPE html>
|
||
|
|
<html lang="en">
|
||
|
|
<head>
|
||
|
|
<meta charset="UTF-8">
|
||
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||
|
|
<title>VAT Company Lookup</title>
|
||
|
|
<style>
|
||
|
|
@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700&display=swap');
|
||
|
|
|
||
|
|
:root {
|
||
|
|
--bg1: #eef6ff;
|
||
|
|
--bg2: #dcecff;
|
||
|
|
--panel: rgba(255, 255, 255, 0.9);
|
||
|
|
--blue: #1f5fae;
|
||
|
|
--blue-dark: #123d73;
|
||
|
|
--blue-soft: #6fa6e8;
|
||
|
|
--line: #4d8fdb;
|
||
|
|
--text: #1a2f4d;
|
||
|
|
--muted: #4f6f92;
|
||
|
|
--shadow: 0 18px 50px rgba(20, 71, 132, 0.16);
|
||
|
|
}
|
||
|
|
|
||
|
|
* {
|
||
|
|
box-sizing: border-box;
|
||
|
|
}
|
||
|
|
|
||
|
|
body {
|
||
|
|
margin: 0;
|
||
|
|
min-height: 100vh;
|
||
|
|
font-family: 'Inter', sans-serif;
|
||
|
|
font-size: 24px;
|
||
|
|
color: var(--text);
|
||
|
|
background:
|
||
|
|
radial-gradient(circle at top left, #f7fbff 0%, transparent 32%),
|
||
|
|
linear-gradient(135deg, var(--bg1), var(--bg2));
|
||
|
|
display: flex;
|
||
|
|
justify-content: center;
|
||
|
|
align-items: flex-start;
|
||
|
|
padding: 48px 24px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.container {
|
||
|
|
width: 100%;
|
||
|
|
max-width: 1100px;
|
||
|
|
background: var(--panel);
|
||
|
|
backdrop-filter: blur(8px);
|
||
|
|
border-radius: 28px;
|
||
|
|
box-shadow: var(--shadow);
|
||
|
|
padding: 40px 48px;
|
||
|
|
}
|
||
|
|
|
||
|
|
h1 {
|
||
|
|
margin: 0 0 36px 0;
|
||
|
|
color: var(--blue-dark);
|
||
|
|
font-size: 34px;
|
||
|
|
font-weight: 700;
|
||
|
|
letter-spacing: -0.5px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.form-grid {
|
||
|
|
display: flex;
|
||
|
|
flex-direction: column;
|
||
|
|
gap: 28px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.row {
|
||
|
|
display: grid;
|
||
|
|
grid-template-columns: 220px 1fr 1fr auto;
|
||
|
|
gap: 20px;
|
||
|
|
align-items: end;
|
||
|
|
}
|
||
|
|
|
||
|
|
label {
|
||
|
|
font-weight: 600;
|
||
|
|
color: var(--blue);
|
||
|
|
padding-bottom: 8px;
|
||
|
|
}
|
||
|
|
|
||
|
|
input, select {
|
||
|
|
width: 100%;
|
||
|
|
font-family: inherit;
|
||
|
|
font-size: 24px;
|
||
|
|
color: var(--text);
|
||
|
|
padding: 10px 4px;
|
||
|
|
border: none;
|
||
|
|
border-bottom: 2px solid var(--blue-soft);
|
||
|
|
background: transparent;
|
||
|
|
outline: none;
|
||
|
|
transition: border-color 0.2s ease;
|
||
|
|
border-radius: 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
input:focus, select:focus {
|
||
|
|
border-bottom-color: var(--blue);
|
||
|
|
box-shadow: none;
|
||
|
|
}
|
||
|
|
|
||
|
|
input::placeholder {
|
||
|
|
color: #7f9fc4;
|
||
|
|
}
|
||
|
|
|
||
|
|
button {
|
||
|
|
font-family: inherit;
|
||
|
|
font-size: 24px;
|
||
|
|
font-weight: 700;
|
||
|
|
padding: 14px 24px;
|
||
|
|
border: none;
|
||
|
|
border-radius: 16px;
|
||
|
|
cursor: pointer;
|
||
|
|
background: linear-gradient(135deg, #3b82d6, #1f5fae);
|
||
|
|
color: white;
|
||
|
|
box-shadow: 0 10px 24px rgba(31, 95, 174, 0.24);
|
||
|
|
transition: transform 0.12s ease, box-shadow 0.12s ease;
|
||
|
|
white-space: nowrap;
|
||
|
|
}
|
||
|
|
|
||
|
|
button:hover {
|
||
|
|
transform: translateY(-1px);
|
||
|
|
box-shadow: 0 14px 28px rgba(31, 95, 174, 0.28);
|
||
|
|
}
|
||
|
|
|
||
|
|
button:active {
|
||
|
|
transform: translateY(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
.button-row {
|
||
|
|
margin-top: 36px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.status {
|
||
|
|
margin-top: 28px;
|
||
|
|
padding: 20px 24px;
|
||
|
|
border-radius: 18px;
|
||
|
|
background: linear-gradient(135deg, rgba(204, 227, 255, 0.7), rgba(232, 243, 255, 0.95));
|
||
|
|
color: var(--muted);
|
||
|
|
white-space: pre-wrap;
|
||
|
|
font-size: 22px;
|
||
|
|
border: 1px solid rgba(77, 143, 219, 0.18);
|
||
|
|
}
|
||
|
|
|
||
|
|
@media (max-width: 900px) {
|
||
|
|
body {
|
||
|
|
padding: 20px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.container {
|
||
|
|
padding: 28px 24px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.row {
|
||
|
|
grid-template-columns: 1fr;
|
||
|
|
gap: 14px;
|
||
|
|
}
|
||
|
|
|
||
|
|
h1 {
|
||
|
|
font-size: 30px;
|
||
|
|
}
|
||
|
|
|
||
|
|
input, select, button {
|
||
|
|
font-size: 20px;
|
||
|
|
}
|
||
|
|
|
||
|
|
.status {
|
||
|
|
font-size: 18px;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
</style>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<div class="container">
|
||
|
|
<h1>VAT Company Lookup</h1>
|
||
|
|
|
||
|
|
<div class="form-grid">
|
||
|
|
<div class="row">
|
||
|
|
<label for="country">Country / VAT</label>
|
||
|
|
<select id="country">
|
||
|
|
<option value="BE">Belgium</option>
|
||
|
|
<option value="HR">Croatia</option>
|
||
|
|
<option value="PL">Poland</option>
|
||
|
|
</select>
|
||
|
|
<input id="vat_number" type="text" placeholder="VAT number">
|
||
|
|
<button type="button" onclick="searchCompany()">Search</button>
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="row">
|
||
|
|
<label for="company_name">Company</label>
|
||
|
|
<input id="company_name" type="text" style="grid-column: span 3;">
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="row">
|
||
|
|
<label for="street">Street</label>
|
||
|
|
<input id="street" type="text" style="grid-column: span 3;">
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="row">
|
||
|
|
<label for="zip">ZIP / City</label>
|
||
|
|
<input id="zip" type="text" placeholder="ZIP">
|
||
|
|
<input id="city" type="text" placeholder="City">
|
||
|
|
<div></div>
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="row">
|
||
|
|
<label for="email">Email</label>
|
||
|
|
<input id="email" type="email" placeholder="example@company.com" style="grid-column: span 3;">
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="row">
|
||
|
|
<label for="phone">Phone</label>
|
||
|
|
<input id="phone" type="text" placeholder="+32 123 45 67 89" style="grid-column: span 3;">
|
||
|
|
</div>
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div class="button-row">
|
||
|
|
<button type="button" onclick="saveData()">Save</button>
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<div id="status" class="status">Ready.</div>
|
||
|
|
</div>
|
||
|
|
|
||
|
|
<script>
|
||
|
|
async function searchCompany() {
|
||
|
|
const country = document.getElementById("country").value;
|
||
|
|
const vatNumber = document.getElementById("vat_number").value.trim();
|
||
|
|
const status = document.getElementById("status");
|
||
|
|
|
||
|
|
if (!country || !vatNumber) {
|
||
|
|
status.textContent = "Please enter a country and VAT number.";
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
status.textContent = "Searching...";
|
||
|
|
|
||
|
|
try {
|
||
|
|
const response = await fetch(`/api/company_lookup?country=${encodeURIComponent(country)}&vat_number=${encodeURIComponent(vatNumber)}`);
|
||
|
|
const data = await response.json();
|
||
|
|
|
||
|
|
if (!response.ok) {
|
||
|
|
status.textContent = data.error || "Lookup failed.";
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
document.getElementById("company_name").value = data.name || "";
|
||
|
|
document.getElementById("street").value = data.street || "";
|
||
|
|
document.getElementById("zip").value = data.zip || "";
|
||
|
|
document.getElementById("city").value = data.city || "";
|
||
|
|
|
||
|
|
let msg = data.found ? `Found via ${data.source}.` : "No company found.";
|
||
|
|
|
||
|
|
if (data.warning) {
|
||
|
|
msg += `\\nWarning: ${data.warning}`;
|
||
|
|
}
|
||
|
|
if (data.fallback_error) {
|
||
|
|
msg += `\\nFallback error: ${data.fallback_error}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
status.textContent = msg;
|
||
|
|
} catch (err) {
|
||
|
|
status.textContent = "Request failed: " + err;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function saveData() {
|
||
|
|
const payload = {
|
||
|
|
country: document.getElementById("country").value,
|
||
|
|
vat_number: document.getElementById("vat_number").value.trim(),
|
||
|
|
company_name: document.getElementById("company_name").value.trim(),
|
||
|
|
street: document.getElementById("street").value.trim(),
|
||
|
|
zip: document.getElementById("zip").value.trim(),
|
||
|
|
city: document.getElementById("city").value.trim(),
|
||
|
|
email: document.getElementById("email").value.trim(),
|
||
|
|
phone: document.getElementById("phone").value.trim()
|
||
|
|
};
|
||
|
|
|
||
|
|
alert(JSON.stringify(payload, null, 2));
|
||
|
|
}
|
||
|
|
</script>
|
||
|
|
</body>
|
||
|
|
</html>
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/")
|
||
|
|
def index():
|
||
|
|
return render_template_string(HTML)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
app.run(debug=True)
|