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

Solved header bug

Browse Source
This commit is contained in:
Kevin Veen-Birkenbach 2023-12-25 23:36:21 +01:00
parent 8c4785dfe6
commit bcc8a7fb00
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
database_entry_seeder.py
View File

@ -1,11 +1,14 @@
import pandas as pd
import argparse
import os
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:
# Check if the file exists and is not empty
if os.path.exists(file_path) and os.path.getsize(file_path) > 0:
# Read the existing CSV file with header
df = pd.read_csv(file_path, sep=';')
else:
# Create a new DataFrame with columns if file does not exist
df = pd.DataFrame(columns=['host', 'database', 'username', 'password'])
# Check if the entry exists and remove it