summaryrefslogtreecommitdiff
path: root/django/db/backends/sqlite3/base.py
diff options
context:
space:
mode:
authorNick Pope <nick.pope@flightdataservices.com>2018-08-29 21:54:38 +0100
committerTim Graham <timograham@gmail.com>2018-09-10 14:41:58 -0400
commit76dfa834e7ceeca97cd8e3cfa86651a955aa3f0c (patch)
treefbcbbb048cde6df3d7feeb75272044d5afe22925 /django/db/backends/sqlite3/base.py
parentaf7a758dcbbcb228ef526551ffd92fea8ae70f21 (diff)
Combined two identical SQLite functions.
Diffstat (limited to 'django/db/backends/sqlite3/base.py')
-rw-r--r--django/db/backends/sqlite3/base.py20
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