mirror of
				https://github.com/kevinveenbirkenbach/docker-volume-backup.git
				synced 2025-11-04 02:27:56 +00:00 
			
		
		
		
	Added database seeder
This commit is contained in:
		
							
								
								
									
										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()
 | 
			
		||||
		Reference in New Issue
	
	Block a user