Add get_total_asof_block_number

app/data_products.py

@@ -22,6 +22,8 @@ def name(self):
 
     def __init__(self):
 
+        self.rx_event_cnt = 0
+
         self.change = defaultdict(dict)
 
         self.total = []
@@ -32,9 +34,12 @@ def __init__(self):
 
     def handle(self, event):
 
-        self.change = event['diff']
+        if int(event['block_number']) not in self.history_bn_to_pos:
+
+            self.rx_event_cnt += 1
 
-        if event['block_number'] not in self.history_bn_to_pos:
+            self.change = defaultdict(dict)
+            self.change.update(**event['diff'])
 
             assert event['timestamp'] not in self.history_ts_to_pos, f"Duplicate timestamp: {event['timestamp']}"
 
@@ -73,6 +78,30 @@ def get_user_vp_at_block(self, address, block_number):
             return 0
 
         return self.history[self.history_bn_to_pos[snapshot_bn]].get(address, 0)
+
+    def get_total_asof_block_number(self, block_number):
+
+        snapshot_bn = self.block_number_to_snapshot_block_number(block_number)
+
+        if snapshot_bn == 0:
+            return 0
+
+        pos = self.history_bn_to_pos[snapshot_bn]
+
+        return self.total[pos]
+
+    def to_dict(self):
+
+        return {
+            'rx_event_cnt' : self.rx_event_cnt,
+            'history': self.history,
+            'total': self.total,
+            'history_bn_to_pos': self.history_bn_to_pos,
+            'history_ts_to_pos': self.history_ts_to_pos,
+            'change': self.change,
+            'history_len': self.history_len,
+
+        }
 
 
 

app/server.py

@@ -1435,12 +1435,7 @@ async def non_ivotes_total_at_block(request, block_number):
 
         non_ivotes = app.ctx.non_ivotes_vp
 
-        snapshot_bn = non_ivotes.block_number_to_snapshot_block_number(int(block_number))
-
-        if snapshot_bn == 0:
-            total_vp = 0
-        else:
-            total_vp = non_ivotes.total[snapshot_bn]
+        total_vp = non_ivotes.get_total_asof_block_number(int(block_number))
 
         return json({
             'block_number': str(block_number),
@@ -1502,6 +1497,17 @@ async def non_ivotes_all_at_block(request, block_number):
         })
 
 
+######
+#
+# Intentionally Undocumented Endpoints (do not use in production)
+#
+#####
+
+    @app.route('/v1/nonivotes_data_dump')
+    @measure
+    async def non_ivotes_data_dumpt(request):
+
+        return json(app.ctx.non_ivotes_vp.to_dict())
 
 #################################################################################
 #