extern/WireGuard-Config-Generator
1
0
Fork 1
mirror of https://github.com/hatamiarash7/WireGuard-Config-Generator.git synced 2025-06-08 02:36:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Lint code

Browse Source
This commit is contained in:
Arash Hatami 2023-03-03 21:44:21 +03:30
parent a5c0502106
commit 1cc69bd516
No known key found for this signature in database
GPG Key ID: D3D9E8CB2E49731F
1 changed files with 23 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
main.py
View File

@ -2,9 +2,9 @@
Generate WireGuard tunnel configuration files from your data
"""
from typing import List
import json
import environ
from typing import List
# Load environment variables from .env file
env = environ.Env()
@ -13,30 +13,31 @@ environ.Env.read_env()
def get_ips() -> List[str]:
"""
Load the list of IP addresses from the `ip-list.json` file and return a list of unique IP addresses.
Load the list of IP addresses from the `ip-list.json` file
and return a list of unique IP addresses.
Returns:
A list of unique IP addresses.
"""
with open('ip-list.json', 'r', encoding='UTF-8') as f:
with open('ip-list.json', 'r', encoding='UTF-8') as ip_list:
ips = []
# Load IP addresses from JSON file
lists = json.load(f)
lists = json.load(ip_list)
# Iterate over groups of IP addresses
for group in lists:
# Iterate over individual IP addresses
for ip in lists[group]:
ips.append(ip)
for endpoint_ip in lists[group]:
ips.append(endpoint_ip)
# Return a list of unique IP addresses
return list(set(ips))
def generate_config(endpoint: dict) -> None:
def generate_config(profile: dict) -> None:
"""
Generate a WireGuard configuration file for the specified endpoint.
Args:
endpoint: A dictionary containing the endpoint name and address.
profile: A dictionary containing the endpoint name and address.
Returns:
None
@ -44,29 +45,29 @@ def generate_config(endpoint: dict) -> None:
# Get the list of unique IP addresses
ips = ", ".join(map(str, get_ips()))
# Define the filename for the configuration file
filename = endpoint['name'] + '.conf'
filename = profile['name'] + '.conf'
# Get environment variables with default values
address = env('ADDRESS', default='10.0.0.1/24')
mtu = env('MTU', default='1420')
persistent_keepalive = env('PERSISTENT_KEEPALIVE', default='25')
keepalive = env('PERSISTENT_KEEPALIVE', default='25')
# Write the configuration file
with open(filename, 'w', encoding='UTF-8') as f:
with open(filename, 'w', encoding='UTF-8') as profile_file:
# Write the [Interface] section
f.write('[Interface]\n')
f.write('PrivateKey = {}\n'.format(env('PRIVATE_KEY')))
f.write('Address = {}\n'.format(address))
f.write('MTU = {}\n\n'.format(mtu))
profile_file.write("[Interface]\n")
profile_file.write(f"PrivateKey = {env('PRIVATE_KEY')}\n")
profile_file.write(f"Address = {address}\n")
profile_file.write(f"MTU = {mtu}\n\n")
# Write the [Peer] section
f.write('[Peer]\n')
f.write('PublicKey = {}\n'.format(env('PUBLIC_KEY')))
f.write('AllowedIPs = {}\n'.format(ips))
f.write('Endpoint = {}\n'.format(endpoint['address']))
f.write('PersistentKeepalive = {}\n'.format(persistent_keepalive))
profile_file.write("[Peer]\n")
profile_file.write(f"PublicKey = {env('PUBLIC_KEY')}\n")
profile_file.write(f"AllowedIPs = {ips}\n")
profile_file.write(f"Endpoint = {endpoint['address']}\n")
profile_file.write(f"PersistentKeepalive = {keepalive}\n")
# Load endpoint data from JSON file
with open('endpoints.json', 'r', encoding='UTF-8') as f:
endpoints = json.load(f)
with open('endpoints.json', 'r', encoding='UTF-8') as endpoints_file:
endpoints = json.load(endpoints_file)
# Generate a configuration file for each endpoint
for endpoint in endpoints:
generate_config(endpoint)