1 files changed, 91 insertions, 0 deletions
diff --git a/entities/snaks.py b/entities/snaks.py
new file mode 100755
index 0000000..8874329
--- /dev/null
+++ b/entities/snaks.py
@@ -0,0 +1,91 @@
+from datetime import datetime
+
+from libs import JsonSerializable
+
+
+class BasicSnak(JsonSerializable):
+    def __init__(self, snaktype, property, datatype, datavalue):
+        self.snaktype = snaktype
+        self.property = property
+        self.datatype = datatype
+        self.datavalue = datavalue
+
+
+class StringSnak(BasicSnak):
+    def __init__(self, property, value):
+        datavalue = {
+            "type": "string",
+            "value": value
+        }
+        super().__init__(snaktype="value", property=property, datatype="string", datavalue=datavalue)
+
+
+class MonoLingualStringSnak(BasicSnak):
+    def __init__(self, property, value, language):
+        datavalue = {
+            "type": "monolingualtext",
+            "value": {
+                "language": language,
+                "text": value
+            }
+        }
+        super().__init__(snaktype="value", property=property, datatype="monolingualtext", datavalue=datavalue)
+
+
+class EntityIdSnak(BasicSnak):
+    def __init__(self, property, entity_type, entity_id):
+        datavalue = {
+            "value": {
+                "entity-type": entity_type,
+                "numeric-id": entity_id
+            },
+            "type": "wikibase-item"
+        }
+        super().__init__(snaktype="value", property=property, datatype="wikibase-entityid", datavalue=datavalue)
+
+
+class UrlSnak(BasicSnak):
+    def __init__(self, property, url):
+        datavalue = {
+            "type": "string",
+            "value": url
+        }
+        super().__init__(snaktype="value", property=property, datatype="url", datavalue=datavalue)
+
+
+class TimeSnak(BasicSnak):
+    def __init__(self, property, date, precision=11):
+        if not isinstance(date, datetime):
+            date = datetime(date)
+        datavalue = {
+            "value": {
+                "time": date.isoformat(),
+                "timezone": 0,
+                "before": 0,
+                "after": 0,
+                "precision": precision,
+                "calendarmodel": "http:\/\/www.wikidata.org\/entity\/Q1985727"
+            },
+            "type": "time"
+        }
+        super().__init__(snaktype="value", property=property, datatype="time", datavalue=datavalue)
+
+
+class GeoSnak(BasicSnak):
+    def __init__(self, latitude, longitude, precision):
+        datavalue = {
+            "value": {
+                "latitude": latitude,
+                "longitude": longitude,
+                "altitude": None,
+                "precision": precision,
+                "globe": "http:\/\/www.wikidata.org\/entity\/Q2"
+            },
+            "type": "globecoordinate"
+        }
+        super().__init__(snaktype="value", property=property, datatype="globe-coordinate", datavalue=datavalue)
+
+
+class SomeValueSnak(BasicSnak):
+    def __init__(self, property):
+        super().__init__(snaktype="somevalue", property=property, datatype=None, datavalue=None)