mirror of
https://github.com/kevinveenbirkenbach/docker-volume-backup.git
synced 2024-11-25 01:31:03 +01:00
Added database seeder
This commit is contained in:
parent
d1f942bc58
commit
d4799af904
36
database_entry_seeder.py
Normal file
36
database_entry_seeder.py
Normal file
@ -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()
|
Loading…
Reference in New Issue
Block a user