From d4799af90472058949b25b9e0108d03e41642dcb Mon Sep 17 00:00:00 2001
From: Kevin Veen-Birkenbach <kevin@veen.world>
Date: Mon, 25 Dec 2023 23:20:30 +0100
Subject: [PATCH] Added database seeder

---
 database_entry_seeder.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 database_entry_seeder.py

diff --git a/database_entry_seeder.py b/database_entry_seeder.py
new file mode 100644
index 0000000..8175ee8
--- /dev/null
+++ b/database_entry_seeder.py
@@ -0,0 +1,36 @@
+import pandas as pd
+import argparse
+
+def check_and_add_entry(file_path, host, database, username, password):
+    # Read the existing CSV file
+    try:
+        df = pd.read_csv(file_path, sep=';', header=None, names=['host', 'database', 'username', 'password'])
+    except FileNotFoundError:
+        df = pd.DataFrame(columns=['host', 'database', 'username', 'password'])
+
+    # Check if the entry exists
+    mask = (df['host'] == host) & (df['database'] == database) & (df['username'] == username) & (df['password'] == password)
+    if not df[mask].empty:
+        print("Entry already exists.")
+    else:
+        # Add the new entry
+        new_entry = {'host': host, 'database': database, 'username': username, 'password': password}
+        df = df.append(new_entry, ignore_index=True)
+        # Save the updated CSV file
+        df.to_csv(file_path, sep=';', index=False)
+        print("New entry added.")
+
+def main():
+    parser = argparse.ArgumentParser(description="Check and add a database entry to a CSV file.")
+    parser.add_argument("file_path", help="Path to the CSV file")
+    parser.add_argument("host", help="Database host")
+    parser.add_argument("database", help="Database name")
+    parser.add_argument("username", help="Username")
+    parser.add_argument("password", help="Password")
+
+    args = parser.parse_args()
+
+    check_and_add_entry(args.file_path, args.host, args.database, args.username, args.password)
+
+if __name__ == "__main__":
+    main()