Rogerer/Expire.py at master · TheHolyRogerCoin/Rogerer · GitHub

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
69
70
71
72
73
74
75
76
77
78
# coding=utf8
import sys, time, threading, traceback

import Global, Transactions, Config, Irc, Logger

if Global.svsevent != None:
	Global.svsevent.set()
Global.svsevent = threading.Event()

def expirer(event):
	try:
		while not event.is_set():
			cur = Transactions.database().cursor()
			cur.execute("SELECT account FROM accounts WHERE (((last_seen < EXTRACT(EPOCH FROM CURRENT_TIMESTAMP - interval '10 weeks')) IS NOT FALSE AND (last_check < EXTRACT(EPOCH FROM CURRENT_TIMESTAMP - interval '6 hours')) IS NOT FALSE) OR ((last_seen > EXTRACT(EPOCH FROM CURRENT_TIMESTAMP - interval '10 weeks')) IS NOT FALSE AND (last_check < EXTRACT(EPOCH FROM CURRENT_TIMESTAMP - interval '1 week')) IS NOT FALSE)) AND NOT EXISTS (SELECT * FROM locked WHERE locked.account = accounts.account) AND account NOT LIKE '@%' ORDER BY COALESCE(last_check, 0) ASC, balance DESC LIMIT 1")
			if cur.rowcount and Config.config["svs"] in Global.instances:
				Irc.instance_send(Config.config["svs"], ("NS", "INFO " + Irc.sanitize_nickname(cur.fetchone()[0])))
			if not event.is_set(): time.sleep(10)
			if not event.is_set(): time.sleep(10)
			if not event.is_set(): time.sleep(5)
	except Exception as e:
		type, value, tb = sys.exc_info()
		Logger.log("te", "ERROR in expirer")
		Logger.log("te", repr(e))
		Logger.log("te", "".join(traceback.format_tb(tb)))
		Logger.irclog("Error in expirer: %s" % (repr(e)))
		Logger.irclog("".join(traceback.format_tb(tb)).replace("\n", " || "))
		del tb
		Global.svsevent.set()
	for instance in Global.instances:
		Global.instances[instance].expirer_dead.set()
threading.Thread(target = expirer, args = (Global.svsevent,)).start()

def bump_last(ilike, t = None):
	if t == None:
		t = int(time.time())
	db = Transactions.database()
	cur = db.cursor()
	cur.execute("UPDATE accounts SET last_seen = %s WHERE account ILIKE %s", (t, ilike))
	db.commit()

def bump_check(ilike):
	db = Transactions.database()
	cur = db.cursor()
	cur.execute("UPDATE accounts SET last_check = EXTRACT(EPOCH FROM CURRENT_TIMESTAMP) WHERE account ILIKE %s", (ilike,))
	db.commit()

def svsdata(data):
	db = Transactions.database()
	cur = db.cursor()
	ilike = data["nick"].replace("\\", "\\\\").replace("_", "\\_")
	cur.execute("SELECT account, last_seen, registered FROM accounts WHERE account ILIKE %s", (ilike,))
	if cur.rowcount:
		acct, last_seen, registered = cur.fetchone()
		bump_check(ilike)
		if "reg" not in data:
			Logger.irclog("Dormant lock on account %s: not registered" % (acct))
			Transactions.lock(acct, True)
		elif registered != None and registered != data["reg"]:
			Logger.irclog("Dormant lock on account %s: registered at %d, remember %d" % (acct, data["reg"], registered))
			Transactions.lock(acct, True)
		else:
			if registered == None:
				cur.execute("SELECT MAX(timestamp) FROM txlog WHERE destination ILIKE %s", (ilike,))
				lasttx = cur.fetchone()[0] if cur.rowcount else None
				if lasttx != None and lasttx < data["reg"]:
					Logger.irclog("Dormant lock on account %s: registered at %d, last tx at %d" % (acct, data["reg"], lasttx))
					Transactions.lock(acct, True)
				else:
					cur.execute("UPDATE accounts SET registered = %s WHERE account ILIKE %s", (data["reg"], ilike))
					db.commit()
			if "userlast" in data:
				bump_last(ilike, data["userlast"])
			elif "last" in data:
				bump_last(ilike, data["last"])
			elif "userlastweeks" in data:
				bump_last(ilike, int(time.time()) - (data["userlastweeks"] + 1) * 86400 * 7 - 60)
			elif "lastweeks" in data:
				bump_last(ilike, int(time.time()) - (data["lastweeks"] + 1) * 86400 * 7 - 60)