Skip to content

Commit e835727

Browse files
committed
Pylint fixes, conserving pixels
1 parent 9851eec commit e835727

File tree

2 files changed

+18
-20
lines changed

2 files changed

+18
-20
lines changed

timezone_field/fields.py

+4-11
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import pytz
22

3-
from django.conf import settings
43
from django.db import models
54
from django.utils.encoding import smart_unicode, smart_str
65

@@ -13,11 +12,11 @@ class TimeZoneField(models.CharField):
1312

1413
def __init__(self, *args, **kwargs):
1514
defaults = {
16-
"max_length": 63,
17-
"choices": TimeZoneField.CHOICES,
15+
'max_length': 63,
16+
'choices': TimeZoneField.CHOICES,
1817
}
1918
defaults.update(kwargs)
20-
return super(TimeZoneField, self).__init__(*args, **defaults)
19+
super(TimeZoneField, self).__init__(*args, **defaults)
2120

2221
def validate(self, value, model_instance):
2322
# coerce value back to a string to validate correctly
@@ -37,14 +36,8 @@ def get_prep_value(self, value):
3736
return smart_unicode(value)
3837
return value
3938

40-
def get_db_prep_save(self, value, connection=None):
41-
"""
42-
Prepares the given value for insertion into the database.
43-
"""
44-
return self.get_prep_value(value)
45-
4639
def flatten_data(self, follow, obj=None):
4740
value = self._get_val_from_obj(obj)
4841
if value is None:
49-
value = ""
42+
value = ''
5043
return {self.attname: smart_unicode(value)}

timezone_field/tests.py

+14-9
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,14 @@
77
from timezone_field.fields import TimeZoneField
88

99

10+
PST = 'America/Los_Angeles'
11+
EST = 'America/New_York'
12+
13+
1014
class TestModel(models.Model):
1115
timezone = TimeZoneField()
1216

17+
1318
class TestModelForm(forms.ModelForm):
1419
class Meta:
1520
model = TestModel
@@ -18,33 +23,33 @@ class Meta:
1823
class TimeZoneFieldTestCase(TestCase):
1924

2025
def test_models_modelform_validation(self):
21-
form = TestModelForm({"timezone": "America/Denver"})
26+
form = TestModelForm({'timezone': PST})
2227
self.assertTrue(form.is_valid())
2328

2429
def test_models_modelform_save(self):
25-
form = TestModelForm({"timezone": "America/Denver"})
30+
form = TestModelForm({'timezone': EST})
2631
self.assertTrue(form.is_valid())
2732
form.save()
2833

2934
def test_models_string_value(self):
30-
p = TestModel(timezone="America/Denver")
35+
p = TestModel(timezone=PST)
3136
p.save()
3237
p = TestModel.objects.get(pk=p.pk)
33-
self.assertEqual(p.timezone, pytz.timezone("America/Denver"))
38+
self.assertEqual(p.timezone, pytz.timezone(PST))
3439

3540
def test_models_string_value_lookup(self):
36-
TestModel(timezone="America/Denver").save()
37-
qs = TestModel.objects.filter(timezone="America/Denver")
41+
TestModel(timezone=EST).save()
42+
qs = TestModel.objects.filter(timezone=EST)
3843
self.assertEqual(qs.count(), 1)
3944

4045
def test_models_tz_value(self):
41-
tz = pytz.timezone("America/Denver")
46+
tz = pytz.timezone(PST)
4247
p = TestModel(timezone=tz)
4348
p.save()
4449
p = TestModel.objects.get(pk=p.pk)
4550
self.assertEqual(p.timezone, tz)
4651

4752
def test_models_tz_value_lookup(self):
48-
TestModel(timezone="America/Denver").save()
49-
qs = TestModel.objects.filter(timezone=pytz.timezone("America/Denver"))
53+
TestModel(timezone=PST).save()
54+
qs = TestModel.objects.filter(timezone=pytz.timezone(PST))
5055
self.assertEqual(qs.count(), 1)

0 commit comments

Comments
 (0)