Làm thế nào để biết tên hiện tại của cơ sở dữ liệu ở Django?

Question

Tôi đang viết bài kiểm tra trong dự án django của mình. Còn bây giờ, tôi có hai kết nối cơ sở dữ liệu:

(settings.py) 
DATABASES = { 
    'default': { 
     'ENGINE': 'django.db.backends.postgresql_psycopg2', 
     'NAME': 'db_name' 
     ... 
    }, 
} 

và tùy chỉnh kết nối đến MongoDB:

import sys 
from pymongo import Connection 
from pymongo.errors import ConnectionFailure 
try: 
    connection = Connection(host="localhost", port=27017) 
    db = connection['db_name'] 
    print "Connected successfully(Mongo, db_name)" 
except ConnectionFailure, e: 
    sys.stderr.write("Could not connect to MongoDB: %s" % e) 
    sys.exit(1) 

và tôi muốn biết khi nào dự án của tôi đang chạy qua

python manage.py test myapp 

Bởi vì khi bạn chạy thử nghiệm, django tự động tạo DB riêng (với tên như test_db_name), nhưng trong trường hợp này Mongo sẽ vẫn chạy với db_name. Tôi cố gắng:

import sys 
from pymongo import Connection 
from pymongo.errors import ConnectionFailure 
from django.db import connections 

try: 
    connection = Connection(host="localhost", port=27017) 
    db_name = connections['default'].settings_dict['NAME'] 
    db = connection[db_name] 
    print "Connected successfully(Mongo, %s)" % (db_name,) 
except ConnectionFailure, e: 
    sys.stderr.write("Could not connect to MongoDB: %s" % e) 
    sys.exit(1) 

nhưng nó không hoạt động

Answer 1

Hãy thử đặt sau đây trong tập tin cài đặt của bạn:

import sys 

management_command = sys.argv[1] if len(sys.argv) > 1 else "" 

TESTING = management_command.startswith("test") 

Nếu TESTING là đúng bạn đang chạy với python kiểm tra manage.py myapp.

chỉnh sửa: Bạn có thể đặt bất kỳ nơi đâu, không nhất thiết là tệp cài đặt của bạn. Cung cấp cho nó một đi và xem nó hoạt động như thế nào!

Answer 2

Bạn có thể kiểm tra xem nó trong db.settings:

from django import db 
db.settings.DATABASES['default']['NAME'] 

Để xem các cơ sở dữ liệu sử dụng để lấy một đối tượng cụ thể mà bạn có thể làm:

object._state.db 

này sẽ cung cấp cho bạn chìa khóa cơ sở dữ liệu trong cấu hình, chẳng hạn là 'mặc định', vì vậy nếu bạn có nhiều cơ sở dữ liệu trong cấu hình, bạn có thể kiểm tra đúng.

Khi bạn chạy thử nghiệm, db.settings phải được cập nhật để chứa tên cơ sở dữ liệu thử nghiệm cụ thể.

Answer 3

IMO, cách tốt nhất để làm điều này, là chức năng không có giấy tờ django sau:

>>> from django.db import connection 
>>> connection.vendor 
'postgresql' or 'sqlite' 

Thx để: https://stackoverflow.com/a/18849255/1237092

Answer 4

Câu trả lời có vẻ là lỗi thời.

Trong django 1.6 bạn có thể làm:

from django import db 
print db.connections.databases 

Answer 5

Để có được tên db với các phiên bản Django gần đây (thử với 1.8):

from django.db import connection 
db_name = connection.settings_dict['NAME'] 
# Or alternatively 
# db_name = connection.get_connection_params()['db'] 

Quan tâm của việc đọc giá trị này sau khi khởi tạo, vì vậy rằng nó có giá trị chính xác khi chạy thử nghiệm đơn vị.

Answer 6

gì làm việc tuyệt vời đối với tôi (từ điển hoàn toàn trống rỗng trong trường hợp của tôi bởi vì tôi đang sử dụng một tập tin read_default_file trong settings.py) chỉ được thực hiện truy vấn SQL sau

SELECT DATABASE() 

Answer 7

Tested trong django 1,9

Đi vào vỏ.

./manage.py shell 

Kiểm tra cơ sở dữ liệu của bạn.

from django import db 
db.connections.databases 

Answer 8

Trong Django> = 1.10 phiên bản

Sử dụng:

from django.conf import settings 
db_name = settings.DATABASES['default']['NAME'] 

Answer 9

Trong Django> = 1.11

Sử dụng:

from django import db 
db_name = db.utils.settings.DATABASES['default']['NAME'] 

Answer 10

Sử dụng truy vấn thô mysql thông qua django, cũng có thể là giải pháp.

from django.db import connection 
cursor = connection.cursor() 
cursor.execute('select database()') 
row = cursor.fetchone() 

print row[0]