스마트 컨트랙트용 API 서버 만들기

from fastapi import FastAPI, HTTPException
from web3 import Web3

import json

app = FastAPI()

# Ganache GUI에 배포된 스마트 컨트랙트 주소
contract_address = "스마트 컨트랙트 주소"

# Ganache GUI에서 제공하는 HTTP RPC 엔드포인트
ganache_rpc_url = "http://localhost:8545"

# Web3 객체 생성
web3 = Web3(Web3.HTTPProvider(ganache_rpc_url))

# 스마트 컨트랙트 ABI (Application Binary Interface) - 컴파일된 스마트 컨트랙트의 인터페이스
abi_file_path = r"<simpleVDR.json 파일 경로>"
with open(abi_file_path, "r", encoding="utf-8") as fp:
    contracts = json.load(fp)

contract_abi = contracts.get("abi", [])

# 컨트랙트 객체 생성
contract = web3.eth.contract(address=contract_address, abi=contract_abi)

@app.post("/register")
async def register_did(_did: str, _document: str):
    try:
        # 스마트 컨트랙트의 등록 함수 호출
        transaction = contract.functions.register(_did, _document).transact({'from': web3.eth.accounts[0]})
        web3.eth.waitForTransactionReceipt(transaction)
        return {"message": "String registered successfully"}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@app.get("/resolver/{_did}")
async def resolve_did(_did: str):
    try:
        # 스마트 컨트랙트의 조회 함수 호출
        result = contract.functions.resolve(_did).call()
        return {"registered_string": result}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

import requests

# API 서버 주소 
api_server = "http://127.0.0.1:8000"

def test_register():
    register_url = f"{api_server}/register"
    params = {"_did" : "56789", "_document" : "sample document"}
    resp = requests.post(register_url, params=params)
    response_data = resp.json()
    print(response_data)

def test_resolve():
    did = "56789"
    resolver_url = f"{api_server}/resolve/{did}"
    resp = requests.get(resolver_url)
    response_data = resp.json()
    print(response_data)
    

if __name__ == "__main__":
    test_register()
    test_resolve()