Add URL field (#128)

Author: aminalaee

docs/declaring_models.md

@@ -68,6 +68,7 @@ See `TypeSystem` for [type-specific validation keyword arguments][typesystem-fie
 * `orm.String(max_length)`
 * `orm.Text()`
 * `orm.Time()`
+* `orm.URL(max_length)`
 * `orm.UUID()`
 * `orm.JSON()`
 

orm/__init__.py

@@ -2,6 +2,7 @@
 from orm.exceptions import MultipleMatches, NoMatch
 from orm.fields import (
     JSON,
+    URL,
     UUID,
     BigInteger,
     Boolean,
@@ -44,6 +45,7 @@
     "String",
     "Text",
     "Time",
+    "URL",
     "UUID",
     "Model",
     "ModelRegistry",

orm/fields.py

@@ -250,3 +250,11 @@ def get_validator(self, **kwargs) -> typesystem.Field:
 
     def get_column_type(self):
         return GenericIP()
+
+
+class URL(String):
+    def get_validator(self, **kwargs) -> typesystem.Field:
+        return typesystem.URL(**kwargs)
+
+    def get_column_type(self):
+        return sqlalchemy.String(length=self.validator.max_length)

tests/test_columns.py

@@ -49,6 +49,7 @@ class User(orm.Model):
         "name": orm.String(allow_null=True, max_length=16),
         "email": orm.Email(allow_null=True, max_length=256),
         "ipaddress": orm.IPAddress(allow_null=True),
+        "url": orm.URL(allow_null=True, max_length=2048),
     }
 
 
@@ -101,8 +102,15 @@ async def test_model_crud():
     user = await User.objects.get()
     assert user.email is None
     assert user.ipaddress is None
+    assert user.url is None
 
-    await user.update(ipaddress="192.168.1.1", name="Chris", email="[email protected]")
+    await user.update(
+        ipaddress="192.168.1.1",
+        name="Chris",
+        email="[email protected]",
+        url="https://encode.io",
+    )
 
     user = await User.objects.get()
     assert isinstance(user.ipaddress, (ipaddress.IPv4Address, ipaddress.IPv6Address))
+    assert user.url == "https://encode.io"