120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
import json
|
|
import requests
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import getpass
|
|
|
|
# Try to get HSD_API_KEY from environment variable
|
|
HSD_API_KEY = os.environ.get("HSD_API_KEY")
|
|
if HSD_API_KEY is None:
|
|
raise Exception("Please set HSD_API_KEY environment variable")
|
|
|
|
days_until_expire = 180
|
|
|
|
wallet_list = ["cold","hot","registry","greg","hnsau","renewservice"]
|
|
|
|
def get_domains(key,wallet):
|
|
url = f"http://x:{key}@127.0.0.1:12039/wallet/{wallet}/name?own=true"
|
|
response = requests.get(url)
|
|
if response.status_code != 200:
|
|
return []
|
|
|
|
# Check if the response is valid JSON
|
|
try:
|
|
response.json()
|
|
except json.decoder.JSONDecodeError:
|
|
return []
|
|
|
|
return response.json()
|
|
|
|
|
|
def domains_expiring_soon(key,wallet,silent=False):
|
|
domains = get_domains(key,wallet)
|
|
expiring = []
|
|
for domain in domains:
|
|
days_left = domain["stats"]["daysUntilExpire"]
|
|
if days_left < days_until_expire:
|
|
if not silent:
|
|
print(f"{domain['name']} expires in {days_left} days")
|
|
expiring.append(domain["name"])
|
|
|
|
if len(expiring) > 0:
|
|
with open(f"/tmp/domains_expiring_soon_{wallet}.txt", "w") as f:
|
|
for domain in expiring:
|
|
f.write(f"{domain}\n")
|
|
return expiring
|
|
|
|
if __name__ == "__main__":
|
|
# optional args --wallet <wallet> --days <days> --api_key <api_key>
|
|
parser = argparse.ArgumentParser(description='Check HSD domains for expiration')
|
|
parser.add_argument('--wallet', help='Wallet to check', default="all")
|
|
parser.add_argument('--days', help='Days until expiration', default=180)
|
|
parser.add_argument('--api_key', help='API key', default=HSD_API_KEY)
|
|
parser.add_argument('--renew', help='Renew domains', action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
HSD_API_KEY = args.api_key
|
|
days_until_expire = args.days
|
|
# Try to convert days_until_expire to an integer
|
|
try:
|
|
days_until_expire = int(days_until_expire)
|
|
except ValueError:
|
|
print("Invalid days_until_expire value")
|
|
sys.exit(1)
|
|
|
|
|
|
wallets = args.wallet
|
|
if wallets == "all":
|
|
wallets = wallet_list
|
|
else:
|
|
wallets = [wallets]
|
|
|
|
for wallet in wallets:
|
|
print(f"Checking {wallet} wallet...")
|
|
domains_expiring_soon(HSD_API_KEY,wallet)
|
|
|
|
if args.renew:
|
|
print("Renewing domains...")
|
|
for wallet in wallets:
|
|
domains = domains_expiring_soon(HSD_API_KEY,wallet,silent=True)
|
|
if len(domains) > 0:
|
|
print(f"Renewing {len(domains)} domains for {wallet} wallet...")
|
|
batch = []
|
|
for domain in domains:
|
|
batch.append(f'["RENEW", "{domain}"]')
|
|
|
|
batchTX = "[" + ", ".join(batch) + "]"
|
|
reqcontent = f'{{"method": "sendbatch","params":[ {batchTX} ]}}'
|
|
|
|
# Select wallet
|
|
response = requests.post(f'http://x:{HSD_API_KEY}@127.0.0.1:12039', data='{"method": "selectwallet","params": [ "'+wallet+'" ]}')
|
|
if response.status_code != 200:
|
|
print(f"Error selecting wallet {wallet}: {response.text}")
|
|
continue
|
|
|
|
# Ask user for password
|
|
password = getpass.getpass(prompt=f"Enter password for wallet {wallet}: ")
|
|
if password != "":
|
|
response = requests.post(f'http://x:{HSD_API_KEY}@127.0.0.1:12039', data='{"method": "walletpassphrase","params": [ "'+password+'", 300 ]}')
|
|
if response.status_code != 200:
|
|
print(f"Error setting password for wallet {wallet}: {response.text}")
|
|
continue
|
|
|
|
response = requests.post(f'http://x:{HSD_API_KEY}@127.0.0.1:12039', data=reqcontent)
|
|
|
|
if response.status_code != 200:
|
|
print(f"Error renewing domains for wallet {wallet}: {response.text}")
|
|
continue
|
|
|
|
output = response.json()
|
|
if "error" in output:
|
|
if output["error"]:
|
|
print(f"Error renewing domains for wallet {wallet}: {output['error']}")
|
|
|
|
print("TX: " + output["result"]["hash"])
|
|
|
|
|