File: __init__.py

package info (click to toggle)
drf-haystack 1.9.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 564 kB
  • sloc: python: 2,608; makefile: 147
file content (54 lines) | stat: -rw-r--r-- 1,205 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# -*- coding: utf-8 -*-

from __future__ import absolute_import, unicode_literals

import os
from django.core.exceptions import ImproperlyConfigured

test_runner = None
old_config = None

os.environ["DJANGO_SETTINGS_MODULE"] = "tests.settings"

import django
if hasattr(django, "setup"):
    django.setup()


def _geospatial_support():
    try:
        import geopy
        from haystack.utils.geo import Point
    except (ImportError, ImproperlyConfigured):
        return False
    else:
        return True
geospatial_support = _geospatial_support()


def _restframework_version():
    import rest_framework
    return tuple(map(int, rest_framework.VERSION.split(".")))
restframework_version = _restframework_version()


def _elasticsearch_version():
    import elasticsearch
    return elasticsearch.VERSION
elasticsearch_version = _elasticsearch_version()


def setup():
    from django.test.runner import DiscoverRunner
    global test_runner
    global old_config

    test_runner = DiscoverRunner()
    test_runner.setup_test_environment()
    old_config = test_runner.setup_databases()


def teardown():
    test_runner.teardown_databases(old_config)
    test_runner.teardown_test_environment()