feat: BooleanField for ORM

examples/cpu_usage/collect.py

@@ -15,6 +15,6 @@
     timestamp = datetime.datetime.now()
     print(timestamp)
     db.insert([
-        CPUStats(timestamp=timestamp, cpu_id=cpu_id, cpu_percent=cpu_percent)
+        CPUStats(timestamp=timestamp, cpu_id=cpu_id, cpu_percent=cpu_percent, working_status=True)
         for cpu_id, cpu_percent in enumerate(stats)
     ])

examples/cpu_usage/models.py

@@ -1,11 +1,12 @@
-from infi.clickhouse_orm import Model, DateTimeField, UInt16Field, Float32Field, Memory
+from infi.clickhouse_orm import Model, DateTimeField, UInt16Field, Float32Field, Memory, BooleanField
 
 
 class CPUStats(Model):
 
     timestamp = DateTimeField()
     cpu_id = UInt16Field()
     cpu_percent = Float32Field()
+    working_status = BooleanField()
 
     engine = Memory()
 

src/infi/clickhouse_orm/fields.py

@@ -162,7 +162,25 @@ def validate(self, value):
         if len(value) > self._length:
             raise ValueError('Value of %d bytes is too long for FixedStringField(%d)' % (len(value), self._length))
 
+class BooleanField(Field):
+    # The ClickHouse column type to use
+    db_type = 'Bool'
+    # The default value
+    class_default = False
 
+    def to_python(self, value, timezone_in_use):
+        # Convert valid values to bool
+        if value in (1, '1', True):
+            return True
+        elif value in (0, '0', False):
+            return False
+        else:
+            raise ValueError('Invalid value for BooleanField: %r' % value)
+
+    def to_db_string(self, value, quote=True):
+        # The value was already converted by to_python, so it's a bool
+        return '1' if value else '0'
+
 class DateField(Field):
 
     min_value = datetime.date(1970, 1, 1)