Update mychain.py

mychain.py

@@ -7,6 +7,7 @@ from huggingface_hub import (upload_file,HfApi)
 token_self = os.environ['HF_TOKEN']
 api = HfApi(token=token_self)
 chain_d="chain1.json"
+chain_t="trans1.json"
 
 
 class Blockchain:
@@ -157,8 +158,8 @@ class Blockchain:
 class MyChainSend:
 
     def __init__(self,chain_load,load=None,create=None):
-        global main_chain
-        main_chain=chain_load
+        global main_chain_send
+        main_chain_send=chain_load
         
         self.pending_transactions = []
         if load == None or load=="":
@@ -174,11 +175,11 @@ class MyChainSend:
         self.create_block(proof=1, previous_hash='0',chain_n=create)
     def create_block(self, balance, proof, previous_hash,prev_block=None,chain_r=None,chain_n=None):
         if chain_r=="" or chain_r==None:
-            chain_r=f"{main_chain.split('datasets/',1)[1].split('/raw',1)[0]}"
+            chain_r=f"{main_chain_send.split('datasets/',1)[1].split('/raw',1)[0]}"
         if chain_n !="" and chain_n !=None:
-            chain_n = f"{main_chain.split('main/',1)[1]}{chain_n}.json"
+            chain_n = f"{main_chain_send.split('main/',1)[1]}{chain_n}.json"
         if chain_n=="" or chain_n==None:
-            chain_n=chain_d
+            chain_n="error_send.json"
         #prev_block = len(blockchain.chain)-1
         #previous_block = blockchain.chain[prev_block]            
         block = {'index': len(self.chain) + 1,
@@ -193,17 +194,17 @@ class MyChainSend:
             self.pending_transactions = []
             self.chain.append(block)
             json_object = json.dumps(self.chain, indent=4)
-            with open("tmp.json", "w") as outfile:
+            with open("tmp_send.json", "w") as outfile:
                 outfile.write(json_object)
             try: 
                 api.upload_file(
-                path_or_fileobj="tmp.json",
+                path_or_fileobj="tmp_send.json",
                 path_in_repo=chain_n,
                 repo_id=chain_r,
                 token=token_self,
                 repo_type="dataset",
                 )
-                os.remove("tmp.json")
+                os.remove("tmp_send.json")
         
             except Exception as e:
                 print(e)
@@ -371,7 +372,7 @@ class MyChainRec:
         if chain_n !="" and chain_n !=None:
             chain_n = f"{main_chain_rec.split('main/',1)[1]}{chain_n}.json"
         if chain_n=="" or chain_n==None:
-            chain_n=chain_d
+            chain_n="error_rec.json"
         #prev_block = len(blockchain.chain)-1
         #previous_block = blockchain.chain[prev_block]             
         block = {'index': len(self.chain) + 1,
@@ -385,17 +386,17 @@ class MyChainRec:
             self.pending_transactions = []
             self.chain.append(block)
             json_object = json.dumps(self.chain, indent=4)
-            with open("tmp1.json", "w") as outfile:
+            with open("tmp_rec.json", "w") as outfile:
                 outfile.write(json_object)
             try: 
                 api.upload_file(
-                path_or_fileobj="tmp1.json",
+                path_or_fileobj="tmp_rec.json",
                 path_in_repo=chain_n,
                 repo_id=chain_r,
                 token=token_self,
                 repo_type="dataset",
                 )
-                os.remove("tmp1.json")
+                os.remove("tmp_rec.json")
         
             except Exception as e:
                 print(e)
@@ -491,9 +492,9 @@ class MyChainRec:
 
 class MyChainTrans:
 
-    def __init__(self,chain_load,block_trans,load=None,create=None):
-        global main_chain
-        main_chain=chain_load
+    def __init__(self,chain_load,load=None,create=None):
+        global main_chain_trans
+        main_chain_trans=chain_load
         
         self.pending_transactions = []
         if load == None or load=="":
@@ -507,17 +508,20 @@ class MyChainTrans:
         self.chain = []
         self.pending_transactions = []
         self.create_block(proof=1, previous_hash='0',chain_n=create)
-    def create_block(self, balance, proof, previous_hash,chain_r=None,chain_n=None):
+    def create_block(self, balance, proof, previous_hash,prev_block=None,chain_r=None,chain_n=None):
         if chain_r=="" or chain_r==None:
-            chain_r=f"{main_chain.split('datasets/',1)[1].split('/raw',1)[0]}"
+            chain_r=f"{main_chain_trans.split('datasets/',1)[1].split('/raw',1)[0]}"
         if chain_n !="" and chain_n !=None:
-            chain_n = f"{main_chain.split('main/',1)[1]}{chain_n}.json"
+            chain_n = f"{main_chain_trans.split('main/',1)[1]}{chain_t}.json"
         if chain_n=="" or chain_n==None:
-            chain_n=chain_d
-        block = {'block':blockchain.print_previous_block,
-                'index': len(self.chain) + 1,
+            chain_n="error_send.json"
+        #prev_block = len(blockchain.chain)-1
+        #previous_block = blockchain.chain[prev_block]            
+        block = {'index': len(self.chain) + 1,
                 'timestamp': str(datetime.datetime.now()),
+                'block': len(blockchain.chain)+1,
                 'transactions': self.pending_transactions,
+                'balance': balance, 
                 'proof': proof,
                 'previous_hash': previous_hash}
         if self.block_valid(block) == True:
@@ -525,17 +529,17 @@ class MyChainTrans:
             self.pending_transactions = []
             self.chain.append(block)
             json_object = json.dumps(self.chain, indent=4)
-            with open("tmp.json", "w") as outfile:
+            with open("tmp_trans.json", "w") as outfile:
                 outfile.write(json_object)
             try: 
                 api.upload_file(
-                path_or_fileobj="tmp.json",
+                path_or_fileobj="tmp_trans.json",
                 path_in_repo=chain_n,
                 repo_id=chain_r,
                 token=token_self,
                 repo_type="dataset",
                 )
-                os.remove("tmp.json")
+                os.remove("tmp_trans.json")
         
             except Exception as e:
                 print(e)
@@ -600,6 +604,7 @@ class MyChainTrans:
         return out 
 
     def chain_valid(self, chain):
+        
         previous_block = chain[0]
         block_index = 1 
         out=True