kevinveenbirkenbach/docker-volume-backup
1
0
Fork 0
mirror of https://github.com/kevinveenbirkenbach/docker-volume-backup.git synced 2024-11-22 00:11:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Removed unecessary host from seeder function

Browse Source
This commit is contained in:
Kevin Veen-Birkenbach 2024-01-06 14:23:23 +01:00
parent e2e62c5835
commit bf9986f282
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
database_entry_seeder.py
View File

@ -2,17 +2,17 @@ import pandas as pd
import argparse import argparse
import os import os
def check_and_add_entry(file_path, instance, host, database, username, password): def check_and_add_entry(file_path, instance, database, username, password):
# Check if the file exists and is not empty # Check if the file exists and is not empty
if os.path.exists(file_path) and os.path.getsize(file_path) > 0: if os.path.exists(file_path) and os.path.getsize(file_path) > 0:
# Read the existing CSV file with header # Read the existing CSV file with header
df = pd.read_csv(file_path, sep=';') df = pd.read_csv(file_path, sep=';')
else: else:
# Create a new DataFrame with columns if file does not exist # Create a new DataFrame with columns if file does not exist
df = pd.DataFrame(columns=['instance','host', 'database', 'username', 'password']) df = pd.DataFrame(columns=['instance', 'database', 'username', 'password'])
# Check if the entry exists and remove it # Check if the entry exists and remove it
mask = (df['instance'] == instance) & (df['host'] == host) & (df['database'] == database) & (df['username'] == username) mask = (df['instance'] == instance) & (df['database'] == database) & (df['username'] == username)
if not df[mask].empty: if not df[mask].empty:
print("Replacing existing entry.") print("Replacing existing entry.")
df = df[~mask] df = df[~mask]
@ -20,7 +20,7 @@ def check_and_add_entry(file_path, instance, host, database, username, password)
print("Adding new entry.") print("Adding new entry.")
# Create a new DataFrame for the new entry # Create a new DataFrame for the new entry
new_entry = pd.DataFrame([{'instance': instance, 'host': host, 'database': database, 'username': username, 'password': password}]) new_entry = pd.DataFrame([{'instance': instance, 'database': database, 'username': username, 'password': password}])
# Add (or replace) the entry using concat # Add (or replace) the entry using concat
df = pd.concat([df, new_entry], ignore_index=True) df = pd.concat([df, new_entry], ignore_index=True)
@ -32,14 +32,13 @@ def main():
parser = argparse.ArgumentParser(description="Check and replace (or add) a database entry in a CSV file.") parser = argparse.ArgumentParser(description="Check and replace (or add) a database entry in a CSV file.")
parser.add_argument("file_path", help="Path to the CSV file") parser.add_argument("file_path", help="Path to the CSV file")
parser.add_argument("instance", help="Database instance") parser.add_argument("instance", help="Database instance")
parser.add_argument("host", help="Database host")
parser.add_argument("database", help="Database name") parser.add_argument("database", help="Database name")
parser.add_argument("username", help="Username") parser.add_argument("username", help="Username")
parser.add_argument("password", nargs='?', default="", help="Password (optional)") parser.add_argument("password", nargs='?', default="", help="Password (optional)")
args = parser.parse_args() args = parser.parse_args()
check_and_add_entry(args.file_path, args.instance, args.host, args.database, args.username, args.password) check_and_add_entry(args.file_path, args.instance, args.database, args.username, args.password)
if __name__ == "__main__": if __name__ == "__main__":
main() main()