From 2d0582e8adaee34243d2dc1952aaab2b24f77c34 Mon Sep 17 00:00:00 2001
From: snsd0805 <levi900227@gmail.com>
Date: Sat, 11 May 2024 20:46:38 +0800
Subject: [PATCH] feat: scheduler contract's caller for testing

---
 main.py                                   |  30 ++++++++++++++++++++++
 requirements.txt                          |   1 +
 src/__pycache__/scheduler.cpython-310.pyc | Bin 0 -> 1688 bytes
 src/__pycache__/utils.cpython-310.pyc     | Bin 0 -> 332 bytes
 src/scheduler.py                          |  29 +++++++++++++++++++++
 src/utils.py                              |   4 +++
 6 files changed, 64 insertions(+)
 create mode 100644 main.py
 create mode 100644 requirements.txt
 create mode 100644 src/__pycache__/scheduler.cpython-310.pyc
 create mode 100644 src/__pycache__/utils.cpython-310.pyc
 create mode 100644 src/scheduler.py
 create mode 100644 src/utils.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..e7ac84f
--- /dev/null
+++ b/main.py
@@ -0,0 +1,30 @@
+import json
+from web3 import Web3
+from src.scheduler import Scheduler
+
+SCHEDULER_ADDR = "0x13D69Cf7d6CE4218F646B759Dcf334D82c023d8e"
+SCHEDULER_ABI_FILE = "../gpu-contract/abi/Scheduler.abi"
+
+PROVIDER1_KEY = "0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80"
+PROVIDER2_KEY = "0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d"
+PROVIDER3_KEY = "0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a"
+
+CLIENT_KEY = "0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6"
+
+w3 = Web3(Web3.HTTPProvider('http://127.0.0.1:8545'))
+
+
+if __name__ == '__main__':
+    if w3.is_connected():
+        scheduler = Scheduler(w3, SCHEDULER_ADDR, SCHEDULER_ABI_FILE)
+
+        provider1 = w3.eth.account.from_key(PROVIDER1_KEY)
+        provider2 = w3.eth.account.from_key(PROVIDER2_KEY)
+        provider3 = w3.eth.account.from_key(PROVIDER3_KEY)
+        client = w3.eth.account.from_key(CLIENT_KEY)
+
+        print(scheduler.getClusters())
+
+    else:
+        print("cannot connected to the chain")
+
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..2603854
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+web3==6.18.0
diff --git a/src/__pycache__/scheduler.cpython-310.pyc b/src/__pycache__/scheduler.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..6ad8881efaabcf9a1f8a227f22269d0260e626c8
GIT binary patch
literal 1688
zcmZuxOK;;g5GJXo-)^u;n?-?MiY;8A@dAql_E5Azw?%vMp(#*67eW!~SYBCjNh!_-
z@+rA4`UiTjkNr!$_SC=7L(v&34?Cy?4u|A$IP=X%7QJ4Zz?aUxo&N3;@)r(vKL-Y<
z(Daue1Q9eN3mQ?1I?LF?i5yCPB*GEyEfMYwi(KKIk&*v5*pm@8?yvIv74%_N#Dd4?
zX@U#8pAUmmX!?5)l0;OHhzTkk!K5o3=}vtCPH)}F6Tb9CApNN;LeYYmK(s~YmPDcG
ziXQYW(TBqargfIgq^L5f3|qZ2K`exlTAL68;L|jd2_*aQd7$C=Ej0ZCL``eLNl9zx
zhOXV3%*mY1oto9IpdT2lIfB4BY%*dB!#5829k~h}DTK^8vYeSPDe_XqNjdV2t7SHA
z3~+}EP=p}<B&rMDKd0kav5;e(Yw_~U%h%(i5E4YCv>Z><yp+joe6g&)TB_nQ6;h40
zO2+y=&6DMO6LOyBX~{W;q45--hV&D9eei(B$-Rm3!52@339!$r#U3Cn+zZVDatn%u
zfNmH-;B48(sp%YH@#g*}Kz&OE{ex{AuwZr`&Rew$D{a9(u(fw*+o`>I7p(ldaj*7a
z6qNlsxB*VpfpCt=I)riKdr*g4iYpdw7vMJTR22*3<wc%IpjBt7(#yD%{GEJn1^{-h
z0q?ZPe~R^NG*rlL)0$Sfoz!Na<VC7WsorK4cIKn=D$NAn*=Z}<R0?r0hZnNkUFT4g
zyaY}qjWX2fMZRlb91P}}mbu_6UhOUQVd9s^D9xv<7%Da`)UFj*GLh-BG#+FJM-84t
zFlwm|M5c~W3{VU)P*z?=Eeu9NK37ptUhx^AVA`NOc(hZNagFHV_9GC8lzs+cNL|{e
z;lD@jkU<as5xqWsq|g%t6AF9oKhzmCOB@Ue3Q6WboGsZnTf8%L0`~;ZQimW+2U5E$
zg*I*yXW580$>F8g3Gs6nehf{c`ystPd<<yE#a=LY5wY1|UVzdBgI8Hl;3Zwtnr_@J
zS+mls*#pm<iNZX9T`eTrcUh6V8~F{De0eGJQb*xg8LRT=DmT4ENhBb=D##~Wt6Z))
zn5k51^%=TyY)(<v?!GfsU%<+fN|+%ebMN{GJnA_Z*}HrIh3L}j=X-hYnNCJ+d;{`$
zAvrf~&KCtffuMIe|E-F%W<?=_#?hrxeX9Y*ECUo`tni`ViOQD2k_2B!Y#OT<k|TQH
vg}tt$aBhMx+zMWL>l|39l9Q@TGi_f(Yw~|SP55;a<2MMnO|wh7^nm>bnsTQa

literal 0
HcmV?d00001

diff --git a/src/__pycache__/utils.cpython-310.pyc b/src/__pycache__/utils.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..6162ae587efbbd40fc1a006e6dea6e8b118d46c1
GIT binary patch
literal 332
zcmYjN!Ait15S>YbmSqbb{D~Yi>p}6bsNj7sf}$7cHtBAm+k~W5gzjm*`wM#XXZ(O%
z{R<E3q_D8>@!n)6%!G-_vJnuz7MJBM#qXHxf>Cls@?$au3SZ#?6uSYD6iB5l5;~-$
zYmz@BYoUh?zGZ9d@f|;4&F-k56E_J$PZLhSA3h^O<UJ|Y1kMBt+*sWN?sTq15_nnF
zI<V4)B%fEG;Pz<z&r~kWN=x5(H94Q0%EBm3);X_bSv9RLmU3acBkRmlrL>dY6|!rq
z+8^7Okk%$wk2EgsFzhx$;vZ)Rznr<a$xPr7hv#-|pYB3BT^ZHY`eNvhpEQPlp(I3`

literal 0
HcmV?d00001

diff --git a/src/scheduler.py b/src/scheduler.py
new file mode 100644
index 0000000..7c385c2
--- /dev/null
+++ b/src/scheduler.py
@@ -0,0 +1,29 @@
+from web3 import Web3
+from src.utils import load_abi
+
+class Scheduler():
+    def __init__(self, w3: Web3, address: str, abi_file: str):
+        self.w3 = w3
+        abi = load_abi(abi_file)
+        self.contract = self.w3.eth.contract(address=address, abi=abi)
+
+    def registerCluster(self, account, gpu_id: int, gpu_num: int):
+        tx = self.contract.functions.registerCluster(gpu_id, gpu_num).build_transaction({
+            'from': account.address,
+            'nonce': self.w3.eth.get_transaction_count(account.address)
+        })
+        signed_tx = self.w3.eth.account.sign_transaction(tx, private_key=account.key)
+
+        tx_hash = self.w3.eth.send_raw_transaction(signed_tx.rawTransaction)
+        tx_receipt = self.w3.eth.wait_for_transaction_receipt(tx_hash)
+        print(tx_receipt)
+        _hash = tx_receipt['transactionHash']
+        print(self.w3.eth.get_transaction(_hash))
+
+    def getClusters(self):
+        return self.contract.functions.getClusters().call()
+
+    def getStartRunEvent(self):
+        event_filter = self.contract.events.StartRun.create_filter(fromBlock=0)
+        events = event_filter.get_new_entries()
+        print(events)
diff --git a/src/utils.py b/src/utils.py
new file mode 100644
index 0000000..f1e56df
--- /dev/null
+++ b/src/utils.py
@@ -0,0 +1,4 @@
+def load_abi(file):
+    with open(file) as fp:
+        abi = fp.read()
+    return abi