FilesExpand file tree

 master

/

example_utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

68 lines (59 loc) · 3 KB

 master

/

example_utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

68 lines (59 loc) · 3 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

import getpass

import json

import os

import eth_account

from eth_account.signers.local import LocalAccount

from hyperliquid.exchange import Exchange

from hyperliquid.info import Info

def setup(base_url=None, skip_ws=False, perp_dexs=None):

config_path = os.path.join(os.path.dirname(__file__), "config.json")

with open(config_path) as f:

config = json.load(f)

account: LocalAccount = eth_account.Account.from_key(get_secret_key(config))

address = config["account_address"]

if address == "":

address = account.address

print("Running with account address:", address)

if address != account.address:

print("Running with agent address:", account.address)

info = Info(base_url, skip_ws, perp_dexs=perp_dexs)

user_state = info.user_state(address)

spot_user_state = info.spot_user_state(address)

margin_summary = user_state["marginSummary"]

if float(margin_summary["accountValue"]) == 0 and len(spot_user_state["balances"]) == 0:

print("Not running the example because the provided account has no equity.")

url = info.base_url.split(".", 1)[1]

error_string = f"No accountValue:\nIf you think this is a mistake, make sure that {address} has a balance on {url}.\nIf address shown is your API wallet address, update the config to specify the address of your account, not the address of the API wallet."

raise Exception(error_string)

exchange = Exchange(account, base_url, account_address=address, perp_dexs=perp_dexs)

return address, info, exchange

def get_secret_key(config):

if config["secret_key"]:

secret_key = config["secret_key"]

else:

keystore_path = config["keystore_path"]

keystore_path = os.path.expanduser(keystore_path)

if not os.path.isabs(keystore_path):

keystore_path = os.path.join(os.path.dirname(__file__), keystore_path)

if not os.path.exists(keystore_path):

raise FileNotFoundError(f"Keystore file not found: {keystore_path}")

if not os.path.isfile(keystore_path):

raise ValueError(f"Keystore path is not a file: {keystore_path}")

with open(keystore_path) as f:

keystore = json.load(f)

password = getpass.getpass("Enter keystore password: ")

secret_key = eth_account.Account.decrypt(keystore, password)

return secret_key

def setup_multi_sig_wallets():

config_path = os.path.join(os.path.dirname(__file__), "config.json")

with open(config_path) as f:

config = json.load(f)

authorized_user_wallets = []

for wallet_config in config["multi_sig"]["authorized_users"]:

account: LocalAccount = eth_account.Account.from_key(wallet_config["secret_key"])

address = wallet_config["account_address"]

if account.address != address:

raise Exception(f"provided authorized user address {address} does not match private key")

print("loaded authorized user for multi-sig", address)

authorized_user_wallets.append(account)

return authorized_user_wallets