Skip to content

Commit

Permalink
fix: 🐛 Correct container names for postgres or mysql respectively and…
Browse files Browse the repository at this point in the history
… fix underscore vs dashes.
  • Loading branch information
jdrew82 committed May 9, 2024
1 parent 7418658 commit 939ebd9
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,11 +252,11 @@ def db_export(context):
print("Exporting the database as an SQL dump...")
if "docker-compose.mysql.yml" in context.nautobot_docker_compose.compose_files:
export_cmd = 'exec db sh -c "mysqldump -u \${NAUTOBOT_DB_USER} –p \${NAUTOBOT_DB_PASSWORD} \${NAUTOBOT_DB_NAME} nautobot > /tmp/nautobot.sql"' # noqa: W605 pylint: disable=anomalous-backslash-in-string
copy_cmd = f"docker cp {context.nautobot_docker_compose.project_name}-mysql-1:/tmp/nautobot.sql nautobot.sql"
else:
export_cmd = 'exec db sh -c "pg_dump -h localhost -d \${NAUTOBOT_DB_NAME} -U \${NAUTOBOT_DB_USER} > /tmp/nautobot.sql"' # noqa: W605 pylint: disable=anomalous-backslash-in-string
copy_cmd = f"docker cp {context.nautobot_docker_compose.project_name}-postgres-1:/tmp/nautobot.sql nautobot.sql"
docker_compose(context, export_cmd, pty=True)

copy_cmd = f"docker cp {context.nautobot_docker_compose.project_name}_postgres_1:/tmp/nautobot.sql dev/nautobot.sql"
print("Copying the SQL Dump locally...")
context.run(copy_cmd)

Expand All @@ -271,12 +271,13 @@ def db_import(context):
sleep(2)

print("Copying DB Dump to DB container...\n")
copy_cmd = f"docker cp dev/nautobot.sql {context.nautobot_docker_compose.project_name}_postgres_1:/tmp/nautobot.sql"
context.run(copy_cmd)

print("Importing DB...\n")
if "docker-compose.mysql.yml" in context.nautobot_docker_compose.compose_files:
copy_cmd = f"docker cp nautobot.sql {context.nautobot_docker_compose.project_name}-mysql-1:/tmp/nautobot.sql"
import_cmd = 'exec db sh -c "mysql -u \${NAUTOBOT_DB_USER} –p \${NAUTOBOT_DB_PASSWORD} < /tmp/nautobot.sql"' # noqa: W605 pylint: disable=anomalous-backslash-in-string
else:
copy_cmd = f"docker cp nautobot.sql {context.nautobot_docker_compose.project_name}-postgres-1:/tmp/nautobot.sql"
import_cmd = 'exec db sh -c "psql -h localhost -U \${NAUTOBOT_DB_USER} < /tmp/nautobot.sql"' # noqa: W605 pylint: disable=anomalous-backslash-in-string
context.run(copy_cmd)

print("Importing DB...\n")
docker_compose(context, import_cmd, pty=True)

0 comments on commit 939ebd9

Please sign in to comment.