cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-07-23 22:39:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update beat start files with better? SQL statement

Browse Source
This commit is contained in:
Andrew-Chen-Wang 2020-04-14 15:05:53 -04:00
parent dd0e7c0c80
commit 25cd8ea72b
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
{{cookiecutter.project_slug}}/compose/local/django/celery/beat/start
View File

@ -22,17 +22,18 @@ try:
# If not, wait for docker-compose up to migrate all tables. # If not, wait for docker-compose up to migrate all tables.
cur = conn.cursor() cur = conn.cursor()
cur.execute( cur.execute(
"select exists(select * from ${POSTGRES_DB}.tables where table_name=%s)", "select exists(select * from information_schema.tables where table_name=%s)",
('django_celery_beat_periodictask',) ('django_celery_beat_periodictask',)
) )
conn.close() if cur.fetchone()[0] == 1:
cur.close()
sys.exit(0)
else:
cur.close()
sys.exit(-1)
except psycopg2.OperationalError: except psycopg2.OperationalError:
sys.exit(-1) sys.exit(-1)
except psycopg2.errors.UndefinedTable:
conn.close()
sys.exit(-1)
sys.exit(0)
END END
} }
until postgres_ready; do until postgres_ready; do

12
{{cookiecutter.project_slug}}/compose/production/django/celery/beat/start
View File

@ -20,15 +20,17 @@ try:
# If not, wait for docker-compose up to migrate all tables. # If not, wait for docker-compose up to migrate all tables.
cur = conn.cursor() cur = conn.cursor()
cur.execute( cur.execute(
"select exists(select * from ${POSTGRES_DB}.tables where table_name=%s)", "select exists(select * from information_schema.tables where table_name=%s)",
('django_celery_beat_periodictask',) ('django_celery_beat_periodictask',)
) )
conn.close() if cur.fetchone()[0] == 1:
cur.close()
sys.exit(0)
else:
cur.close()
sys.exit(-1)
except psycopg2.OperationalError: except psycopg2.OperationalError:
sys.exit(-1) sys.exit(-1)
except psycopg2.errors.UndefinedTable:
conn.close()
sys.exit(-1)
sys.exit(0) sys.exit(0)
END END