diff options
| author | Nick Pope <nick.pope@flightdataservices.com> | 2018-08-29 21:54:38 +0100 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2018-09-10 14:41:58 -0400 |
| commit | 76dfa834e7ceeca97cd8e3cfa86651a955aa3f0c (patch) | |
| tree | fbcbbb048cde6df3d7feeb75272044d5afe22925 /django/db/backends/sqlite3/base.py | |
| parent | af7a758dcbbcb228ef526551ffd92fea8ae70f21 (diff) | |
Combined two identical SQLite functions.
Diffstat (limited to 'django/db/backends/sqlite3/base.py')
| -rw-r--r-- | django/db/backends/sqlite3/base.py | 20 |
1 files changed, 2 insertions, 18 deletions
diff --git a/django/db/backends/sqlite3/base.py b/django/db/backends/sqlite3/base.py index 6316e2f9da..a6542db1eb 100644 --- a/django/db/backends/sqlite3/base.py +++ b/django/db/backends/sqlite3/base.py @@ -161,7 +161,7 @@ class DatabaseWrapper(BaseDatabaseWrapper): def get_new_connection(self, conn_params): conn = Database.connect(**conn_params) - conn.create_function("django_date_extract", 2, _sqlite_date_extract) + conn.create_function("django_date_extract", 2, _sqlite_datetime_extract) conn.create_function("django_date_trunc", 2, _sqlite_date_trunc) conn.create_function("django_datetime_cast_date", 2, _sqlite_datetime_cast_date) conn.create_function("django_datetime_cast_time", 2, _sqlite_datetime_cast_time) @@ -337,22 +337,6 @@ def _sqlite_datetime_parse(dt, tzname=None): return dt -def _sqlite_date_extract(lookup_type, dt): - dt = _sqlite_datetime_parse(dt) - if dt is None: - return None - if lookup_type == 'week_day': - return (dt.isoweekday() % 7) + 1 - elif lookup_type == 'week': - return dt.isocalendar()[1] - elif lookup_type == 'quarter': - return math.ceil(dt.month / 3) - elif lookup_type == 'iso_year': - return dt.isocalendar()[0] - else: - return getattr(dt, lookup_type) - - def _sqlite_date_trunc(lookup_type, dt): dt = _sqlite_datetime_parse(dt) if dt is None: @@ -398,7 +382,7 @@ def _sqlite_datetime_cast_time(dt, tzname): return dt.time().isoformat() -def _sqlite_datetime_extract(lookup_type, dt, tzname): +def _sqlite_datetime_extract(lookup_type, dt, tzname=None): dt = _sqlite_datetime_parse(dt, tzname) if dt is None: return None |
