Skip to content

Commit c2ff6a8

Browse files
authored
feat: django compatibility field name (#2160)
1 parent c0de49f commit c2ff6a8

2 files changed

Lines changed: 23 additions & 0 deletions

File tree

tests/fields/test_common.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,3 +27,12 @@ async def test_if_has_non_null_default_then_not_required():
2727
@pytest.mark.asyncio
2828
async def test_if_null_default_then_required():
2929
assert fields.TextField(default=None).required is True
30+
31+
32+
def test_django_field_name_compatibility():
33+
assert fields.IntegerField is fields.IntField
34+
assert fields.BigIntegerField is fields.BigIntField
35+
assert fields.SmallIntegerField is fields.SmallIntField
36+
assert fields.DateTimeField is fields.DatetimeField
37+
assert fields.DurationField is fields.TimeDeltaField
38+
assert fields.ForeignKey is fields.ForeignKeyField

tortoise/fields/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,4 +80,18 @@
8080
"OneToOneNullableRelation",
8181
"OneToOneRelation",
8282
"ReverseRelation",
83+
"IntegerField",
84+
"SmallIntegerField",
85+
"DurationField",
86+
"BigIntegerField",
87+
"DateTimeField",
88+
"ForeignKey",
8389
]
90+
91+
# Django field name compatibility
92+
from tortoise.fields.data import BigIntField as BigIntegerField
93+
from tortoise.fields.data import DatetimeField as DateTimeField
94+
from tortoise.fields.data import IntField as IntegerField
95+
from tortoise.fields.data import SmallIntField as SmallIntegerField
96+
from tortoise.fields.data import TimeDeltaField as DurationField
97+
from tortoise.fields.relational import ForeignKeyField as ForeignKey

0 commit comments

Comments
 (0)