From 824e9a8e072af10dfcc3229ec6fc42e4841b20b8 Mon Sep 17 00:00:00 2001
From: mjalalzai <33738574+MForensic@users.noreply.github.com>
Date: Sun, 17 Sep 2023 09:53:49 -0700
Subject: [PATCH] Data classes

---
 carnot/carnot-vote-aggregation.py | 58 +++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 carnot/carnot-vote-aggregation.py

diff --git a/carnot/carnot-vote-aggregation.py b/carnot/carnot-vote-aggregation.py
new file mode 100644
index 0000000..52c9c9f
--- /dev/null
+++ b/carnot/carnot-vote-aggregation.py
@@ -0,0 +1,58 @@
+from dataclasses import dataclass
+from typing import Union, List, Set
+from abc import ABC, abstractmethod
+
+Id = bytes
+View = int
+Committee = Set[Id]
+
+def int_to_id(i: int) -> Id:
+    return bytes(str(i), encoding="utf8")
+
+@dataclass(unsafe_hash=True)
+class StandardQc:
+    block: Id
+    view_num: View  # Changed the variable name to avoid conflict with the class name
+
+    def view(self) -> View:
+        return self.view_num  # Changed the method name to view_num
+
+@dataclass
+class AggregateQc:
+    qcs: List[View]
+    highest_qc: StandardQc
+    view_num: View  # Changed the variable name to avoid conflict with the class name
+
+    def view(self) -> View:
+        return self.view_num  # Changed the method name to view_num
+
+    def high_qc(self) -> StandardQc:
+        assert self.highest_qc.view() == max(self.qcs)  # Corrected method call
+        return self.highest_qc
+
+Qc = Union[StandardQc, AggregateQc]  # Changed the type alias to use Union
+
+@dataclass
+class Block:
+    view_num: View  # Changed the variable name to avoid conflict with the class name
+    qc: Qc
+    _id: Id
+
+    def extends(self, ancestor):
+        if self == ancestor:
+            return True
+        elif self.parent is None:
+            return False
+        elif self.parent.view < ancestor.view:  # Check the view of the parent
+            return False
+        else:
+            return self.parent.extends(ancestor)
+
+    def parent(self) -> Id:
+        if isinstance(self.qc, StandardQc):
+            return self.qc.block
+        elif isinstance(self.qc, AggregateQc):
+            return self.qc.high_qc().block
+
+    def id(self) -> Id:
+        return self._id