tmp.py

#%%
# %load_ext autoreload
# %autoreload 2

import sys
sys.path.append('src/')

import driftpy
print(driftpy.__path__)

from driftpy.constants.config import configs
from anchorpy import Provider
import json 
from anchorpy import Wallet
from solana.rpc.async_api import AsyncClient
from driftpy.drift_client import DriftClient
from driftpy.accounts import get_user_account
from dataclasses import asdict
from solana.keypair import Keypair
import asyncio

async def main():
    global summary_data

    with open('tmp.json', 'r') as f: secret = json.load(f) 
    kp = Keypair.from_secret_key(bytes(secret))
    print('pk:', kp.public_key)

    config = configs['devnet']
    url = 'https://api.devnet.solana.com'
    wallet = Wallet(kp)
    connection = AsyncClient(url)
    provider = Provider(connection, wallet)
    dc = DriftClient.from_config(config, provider)

    while True: 
        print('settling...')
        await dc.settle_lp(
            dc.authority, 
            0
        )
        user = await get_user_account(
            dc.program, 
            kp.public_key,
        )
        position = user.positions[0]
        print('position:', position)

        summary_data.append(
            asdict(position)
        )

        await asyncio.sleep(10)
        print()

summary_data = []
try:
    asyncio.run(main())
finally:
    import pandas as pd 
    df = pd.DataFrame(summary_data)
    df.to_csv("lp_summary_bot.csv", index=False)

#%%
#%%
#%%
# from driftpy.constants.numeric_constants import AMM_RESERVE_PRECISION
# n_tokens = 1_000 * AMM_RESERVE_PRECISION
# sig = await dc.add_liquidity(
#     n_tokens, 0
# )
# sig

# #%%
# import time 

#%%
#%%
# # %%
# pk = PublicKey("2zJhfetddV3J89zRrQ6o9W4KW2JbD6QYHjB7uT2VsgnG")
# drift_user = User(
#     dc, 
#     pk
# )

# %%
# %%