Skip to main content

parsing and writing WireGuard configuration files

Project description

wgconfig

Parsing and writing WireGuard configuration files (comment preserving)

WireGuard config files are ini-style. Since all "Peer" sections have the same name, these files cannot be parsed and modified by most modules handling configuration files. Most existing modules are not able to preserve or even add comments when modifying a config file. "wgconfig" was created to work with WireGuard configuration files and to preserve comments.


Features

  • Read and parse WireGuard configuration files and make the data available as Python dictionaries
  • Create new WireGuard configuration files
  • Add peers to WireGuard configuration files and delete peers from WireGuard configuration files
  • Save and clone WireGuard configuration files
  • Comments are preserved when reading and writing WireGuard configuration files
  • Leading comments may be added when creating sections or attributes
  • Such comments may be deleted when removing sections or attributes
  • No other modules are needed, i.e. no dependencies

Installation

Install using PyPi:

pip3 install wgconfig

Quickstart

Reading and parsing an existing WireGuard configuration file

Read and parse the existing WireGuard configuration file 'wg0.conf' located in '/etc/wireguard':

import wgconfig
wc = wgconfig.WGConfig('wg0')
wc.read_file()
print('INTERFACE DATA:', wc.interface)
print('PEER DATA:', wc.peers)

Add a new peer with a comment line before the peer section:

wc.add_peer('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', '# Newly added peer')

Add an attribute to that peer:

wc.add_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'Endpoint', 'wg.example.com:51820', '# Added for demonstration purposes')

Write the changes to disk. Comments that were present when reading the file are preserved.

wc.write_file()

Please see below for more detailed usage information.

Creating a new WireGuard configuration file

Create a new WireGuard configuration file as '/root/wgtest.conf':

import wgconfig
wc = wgconfig.WGConfig('/root/wgtest.conf')
# Add attribute to Interface section (denoted by 'None')
wc.add_attr(None, 'PrivateKey', '6FYKQKEtGFAb5HSwyj5cQl3wgS1E9d6SqVjdVksOn2s=')
# Save to disk
wc.write_file()
# Access the data
print('INTERFACE DATA:', wc.interface)
print('PEER DATA (there are no peers yet):', wc.peers)

The module also contains simple wrappers around the wg command to generate and manage keys:

import wgconfig.wgexec as wgexec
# Create a new WireGuard private key
private_key = wgexec.generate_privatekey()

Please see below for more detailed usage information.


Usage / API

Properties

  • interface: Returns attributes and values of the Interface section as a dictionary

  • peers: Returns attributes and values of all peers as a nested dictionary

Methods for interaction

  • __init__(file)

    Initializes the instance

    Parameters:

    • "file" (str): Path of the WireGuard configuration file You may also just provide the interface name. In this case, the path '/etc/wireguard' is assumed along with a file extension '.conf'.

    Examples:

    • wc = wgconfig.WGConfig('wg0')
    • wc = wgconfig.WGConfig('/etc/wireguard/wg0.conf')
  • read_file()

    Reads the WireGuard config file into memory

  • write_file(file)

    Writes a WireGuard config file from memory to file

    Parameters:

    • "file" (str, optional, default: None): Path of the WireGuard configuration file You may also just provide the interface name. In this case the path '/etc/wireguard' is assumed along with a file extension '.conf'. In case the parameter is missing, the config file defined on object initialization is used.

    Examples:

    • wc.write_file()
    • wc.write_file('wg0')
    • wc.write_file('/etc/wireguard/wg0.conf')
  • initialize_file(leading_comment)

    Empties the file and adds the interface section header

    Parameters:

    • "leading_comment" (str, optional, default: None): Comment line to be added before the Interface section. Must start with a '#' to indicate a comment.

    Examples:

    • wc.initialize_file()
    • wc.initialize_file('# Here comes the Interface section:')
  • add_peer(key, leading_comment)

    Adds a new peer with the given (public) key

    Parameters:

    • "key" (str): Public key of the new peer
    • "leading_comment" (str, optional, default: None): Comment line to be added before the Peer section. Must start with a '#' to indicate a comment.

    Examples:

    • wc.add_peer('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=')
    • wc.add_peer('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', '# Here comes the Interface section:')
  • del_peer(key)

    Removes the peer with the given (public) key

    Note: Comment lines immediately before the Peer section are removed, too.

    Parameters:

    • "key" (str): Public key of the peer

    Examples:

    • wc.del_peer('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=')
  • add_attr(key, attr, value, leading_comment, append_as_line)

    Adds an attribute/value pair to the given peer ('None' for adding an interface attribute)

    Parameters:

    • "key" (str): Key of the peer. Set to 'None' to denote the Interface section
    • "attr" (str) Name of the attribute to add
    • "value" (str or int) Value of the attribute to add
    • "leading_comment" (str, optional, default: None): Comment line to be added before the Peer section. Must start with a '#' to indicate a comment.
    • "append_as_line" (bool, optional, default: False): Whether to add the attribute as a new line if another attribute with the same name already exists. If "False", adding an attribute that already exists results in comma-separated attribute values. This way, "AllowedIPs" can be added one by one.

    Examples:

    • wc.add_attr(None, 'ListenPort', '51820')
    • wc.add_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'AllowedIPs', '0.0.0.0/0')
    • wc.add_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'AllowedIPs', '0.0.0.0/0', '# Allow all IPv4 addresses', append_as_line=True)
  • del_attr(self, key, attr, value, remove_leading_comments)

    Removes an attribute/value pair from the given peer ('None' for adding an interface attribute); set 'value' to 'None' to remove all values

    Parameters:

    • "key" (str): Key of the peer. Set to 'None' to denote the Interface section
    • "attr" (str) Name of the attribute to remove
    • "value" (str or int, optional, default: None) Value of the attribute to remove Set to 'None' if all values (either comma-separated or is multiple attribute lines) shall be removed. Otherwise specify the specific value to be removed.
    • "remove_leading_comments" (bool, optional, default: True): Indicates whether comment lines before the attribute line(s) shall be removed, too

    Examples:

    • wc.del_attr(None, 'ListenPort')
    • wc.del_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'AllowedIPs')
    • wc.del_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'AllowedIPs', '0.0.0.0/0')
    • wc.del_attr('801mgm2JhjTOCxfihEknzFJGYxDvi+8oVYBrWe3hOWM=', 'AllowedIPs', '0.0.0.0/0', remove_leading_comments=False)

Reporting bugs

In case you encounter any bugs, please report the expected behavior and the actual behavior so that the issue can be reproduced and fixed.


Developers

Clone repository

Clone this repo to your local machine using https://github.com/towalink/wgconfig.git

Install the module temporarily to make it available in your Python installation:

pip3 install -e <path to root of "src" directory>

Run unit tests

Call "pytest" to run the unit tests:

pytest <path to root of "test" directory>

License

License

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

wgconfig-0.2.3.tar.gz (22.6 kB view hashes)

Uploaded Source

Built Distribution

wgconfig-0.2.3-py3-none-any.whl (20.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page