def test_at_cancel(self):
job = atd.at("echo lol", datetime.timedelta(seconds=7200))
self.assertGreater(job.id, 0)
print("Success: Created atjob {}".format(job.id))
# Test if we can find our job again by command in `atq`
atq = atd.AtQueue()
n_jobs = len(atq.jobs)
# We should have at least one job in the queue since we just created
# one above ...
self.assertGreater(n_jobs, 0)
print("Success: At least one job in queue ({} jobs)".format(n_jobs))
our_job = None
for j in atq.jobs:
if job.command == "echo lol":
our_job = j
self.assertIsNotNone(our_job)
self.assertEqual(our_job.id, job.id)
print("Success: Found our job again. Job #{}".format(our_job.id))
result = atd.atrm(our_job)
self.assertTrue(result)
print("Success: Deleted our job")
def test_at_queue_schedule(self):
# Clear both atq Q and atq U
atd.clear('Q')
atd.clear('U')
for q in ['Q', 'U']:
for i in range(0, 5):
atd.at("echo", "now + 24 hours", queue=q)
atq = atd.AtQueue('Q')
atq2 = atd.AtQueue('U')
self.assertEqual(len(atq.jobs), len(atq2.jobs))
self.assertEqual(len(atq.jobs), 5)
self.assertEqual(atq2.jobs[0].command.decode("utf-8"), 'echo')
atd.clear('Q')
atd.clear('U')
atq.refresh()
atq2.refresh()
self.assertEqual(len(atq.jobs), len(atq2.jobs))
self.assertEqual(len(atq.jobs), 0)
from atd import atd
config = configparser.ConfigParser()
config.read('/config/config.ini')
pushover_client = pushover.Client(config['pushover']['user_key'],
api_token=config['pushover']['api_token'])
pushover_client.send_message("Starting transfer from NiceHash to CoinSpot",
title="Starting NiceHash to CoinSpot Transfer")
atd.clear()
next_run_date = (datetime.datetime.now().replace(
day=1, hour=0, minute=0, second=0, microsecond=0) +
datetime.timedelta(days=32)).replace(day=1)
backup_job = atd.at("python3 /nicehash_auto_withdraw/nicehash_to_coinspot.py",
next_run_date)
pushover_client.send_message("Set backup job in case of crash",
title="Backup Job")
private_api = nicehash.private_api(config['nicehash']['host'],
config['nicehash']['organisation_id'],
config['nicehash']['api_key'],
config['nicehash']['api_secret'],
config['nicehash'].getboolean('verbose'))
my_btc_account = private_api.get_accounts_for_currency("BTC")
nh_balance = float(my_btc_account['available'])
# print(nh_balance)
pushover_client.send_message("NiceHash confirmed balance: " + str(nh_balance) +
" BTC",