add temporary script for adding ovpn configs to database

1 files changed, 31 insertions, 0 deletions

diff --git a/src/add_config.py b/src/add_config.py
new file mode 100755
index 0000000..4fd4feb
--- /dev/null
+++ b/src/add_config.py
@@ -0,0 +1,31 @@
+#!/bin/python3
+
+from sys import argv
+import yaml
+import psycopg2
+import hashlib
+
+db_config_path = '/etc/0tdns/db_connection_config.yml'
+
+ovpn_config_path = argv[1]
+
+with open(ovpn_config_path) as file:
+    ovpn_config_text = file.read()
+
+ovpn_config_raw = bytearray(ovpn_config_text, encoding='utf-8')
+
+ovpn_config_hash = hashlib.sha256(ovpn_config_raw).hexdigest()
+
+config = yaml.safe_load(open(db_config_path, 'r'))
+connection = psycopg2.connect(user=config['user'], password=config['password'],
+                              host=config['host'], port=config['port'],
+                              database=config['database'])
+cursor = connection.cursor()
+
+cursor.execute('''
+INSERT INTO vpn (location_id, ovpn_config, ovpn_config_sha256)
+VALUES(%s, %s, %s)''', (11, ovpn_config_text, ovpn_config_hash))
+
+connection.commit()
+cursor.close()
+connection.close()