浏览代码

Heroku related

sleepytaco 3 年之前
父节点
当前提交
170325b919
共有 2 个文件被更改,包括 12 次插入0 次删除
  1. 11 0
      UnTube/settings.py
  2. 1 0
      requirements.txt

+ 11 - 0
UnTube/settings.py

@@ -11,6 +11,10 @@ https://docs.djangoproject.com/en/3.2/ref/settings/
 """
 import os
 from pathlib import Path
+
+import django_heroku
+import psycopg2
+
 from UnTube.secrets import SECRETS
 
 
@@ -121,6 +125,9 @@ DATABASES = {
     }
 }
 
+DATABASE_URL = os.environ['DATABASE_URL']
+conn = psycopg2.connect(DATABASE_URL, sslmode='require')
+
 # Password validation
 # https://docs.djangoproject.com/en/3.2/ref/settings/#auth-password-validators
 AUTH_PASSWORD_VALIDATORS = [
@@ -158,3 +165,7 @@ STATIC_ROOT = os.path.join(BASE_DIR, "static")
 # Default primary key field type
 # https://docs.djangoproject.com/en/3.2/ref/settings/#default-auto-field
 DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'
+
+django_heroku.settings(locals())
+import dj_database_url
+DATABASES['default'] = dj_database_url.config(conn_max_age=600, ssl_require=True)

+ 1 - 0
requirements.txt

@@ -25,6 +25,7 @@ oauthlib==3.1.0
 packaging==20.9
 protobuf==3.17.1
 psycopg2==2.8.6
+psycopg2-binary==2.8.6
 pyasn1==0.4.8
 pyasn1-modules==0.2.8
 pycparser==2.20