init scripts

update-nord.py

@@ -0,0 +1,71 @@
+import psycopg2
+from datetime import datetime
+
+# Read database connection details from psql.creds
+db_config = {}
+try:
+    with open('psql.creds', 'r') as f:
+        for line in f:
+            if '=' not in line:
+                continue
+
+            key, value = line.strip().split('=')
+            db_config[key.strip()] = value.strip()
+
+except FileNotFoundError:
+    print("Error: psql.creds file not found. Please make sure it exists in the same directory as the script.")
+    exit(1)
+
+# Initialize counters for the summary
+total_processed = 0
+successful_updates = 0
+
+try:
+    # Connect to the PostgreSQL database using the config
+    conn = psycopg2.connect(
+        dbname=db_config['db_name'],
+        user=db_config['db_user'],
+        password=db_config['db_password'],
+        host=db_config['db_host'],
+        port=db_config['db_port']
+    )
+    cur = conn.cursor()
+
+    # Read the file line by line
+    with open('vpnlist.txt', 'r') as f:
+        for line in f:
+            if line.startswith('#'):
+                continue
+
+            parts = line.strip().split(' # ')
+            exit_ip = parts[0]
+            file_name = parts[1].strip()
+
+            total_processed += 1
+
+            # Update the database
+            current_timestamp = datetime.now()
+            cur.execute(
+                """
+                UPDATE ovpn_files
+                SET exit_ip = %s, last_exit_ip_check = %s
+                WHERE file_name = %s
+                """,
+                (exit_ip, current_timestamp, file_name)
+            )
+
+            successful_updates += cur.rowcount
+
+    # Commit the changes to the database
+    conn.commit()
+
+    # Print the summary
+    print(f"Processed {total_processed} files. Successfully updated {successful_updates} rows in the database.")
+
+except (Exception, psycopg2.DatabaseError) as error:
+    print(f"Error connecting to the database: {error}")
+
+finally:
+    if conn is not None:
+        cur.close()
+        conn.close()