stacks-puppet-node/blockstack_cli/blockstack_registrar/register_daemon.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#-----------------------
# Copyright 2014 Halfmoon Labs, Inc.
# All Rights Reserved
#-----------------------

import os
import json

from config import MONGODB_URI, LOAD_BALANCER, OLD_DB

from coinrpc import namecoind 

from blockdata.register import process_user

from pymongo import MongoClient
from bson.objectid import ObjectId

from encrypt.bip38 import bip38_decrypt

import datetime
import hashlib

from time import sleep

from tools.sweep_btc import sweep_btc

#-----------------------------------
remote_client = MongoClient(MONGODB_URI)
remote_db = remote_client.get_default_database()
users = remote_db.user
registrations = remote_db.user_registration
updates = remote_db.profile_update
transfer = remote_db.name_transfer

old_client = MongoClient(OLD_DB)
old_db = old_client.get_default_database()
old_users = old_db.user

local_client = MongoClient() 
local_db = local_client['namecoin']

from config_local import problem_users, banned_users 
#
#-----------------------------------
def process_profile(username,profile):

	if username in problem_users:
		return

	try:
		process_user(username,profile)
	except Exception as e:
		print e 

#-----------------------------------
def profile_on_blockchain(username,DB_profile):

	sleep(1)
	try:
		block_profile = namecoind.get_full_profile('u/' + username)
	except:
		return False

	block_profile = json.dumps(block_profile,sort_keys=True)
	DB_profile = json.dumps(DB_profile,sort_keys=True)

	if len(block_profile) == len(DB_profile):
		#check hash for only profiles where length is the same
		if hashlib.md5(block_profile).hexdigest() == hashlib.md5(DB_profile).hexdigest():
			return True
		else:
			return False 
	else:
		return False

#-----------------------------------
def check_banned(username):

	for user in banned_users:
		if user in username: 
			return True

	return False 

#-----------------------------------
def register_users(): 

	for new_user in registrations.find():

		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})

		if check_banned(user['username']):
			continue
			
		if 'dispatched' in new_user and new_user['dispatched'] is False: 
	
			if datetime.datetime.utcnow() - new_user['created_at'] > datetime.timedelta(minutes=15):
				print "Dispatch: " + user['username']
				
				process_profile(user['username'],user['profile'])
				new_user['dispatched'] = True 
				registrations.save(new_user)
			else:
				print "New user (within 15 mins): " + user['username']
		
		elif 'dispatched' in new_user and new_user['dispatched'] is True:
		
			try:
				block_profile = namecoind.get_full_profile('u/' + user['username'])
			except:
				print "ERROR: getting block profile: " + str(user['username'])
			
			if profile_on_blockchain(user["username"],user["profile"]):
				registrations.remove(new_user)
			else:
				if datetime.datetime.utcnow() - new_user['created_at'] > datetime.timedelta(minutes=90):
				
					print "Problem (90 mins): " + user['username']
					#print "Re-sending after 180 mins: " + user['username']
					process_profile(user['username'],user['profile'])
			
		else:
			print "Random: " + user['username']
			#registrations.remove(new_user)

#-----------------------------------
def check_users(): 

	counter = 0 

	for user in users.find():

		if check_banned(user['username']):
					continue

		if 'username_activated' in user and user['username_activated'] is False:
			continue

		counter += 1

		print user['username']

		if profile_on_blockchain(user["username"],user["profile"]):
			#print "Fine: " + user["username"]
			pass
		else:
			print "Problem: " + user["username"]
			#process_profile(user['username'],user['profile'])

	print "Users: " + str(counter)

#-----------------------------------
def check_transfer(): 

	for new_user in transfer.find():
	
		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})

		if user is None:
			old_user = old_users.find_one({"_id":user_id})
			user = users.find_one({"username":old_user['username']})

		if check_banned(user['username']):
			continue
		
		if profile_on_blockchain(user["username"],user["profile"]):
			#transfer.remove(new_user)
			pass
		else:
			print "Problem: " + user["username"]
			process_profile(user['username'],user['profile'])

#-----------------------------------
def update_users(): 

	for new_user in updates.find():
		
		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})

		if user is None:
			old_user = old_users.find_one({"_id":user_id})
			user = users.find_one({"username":old_user['username']})

		if check_banned(user['username']):
			continue
		
		if profile_on_blockchain(user["username"],user["profile"]):
			updates.remove(new_user)
		else:
			print "Update: " + str(user['username'])
			process_profile(user['username'],user['profile'])

#-----------------------------------
def cleanup_db(): 

	print "----------"
	print "Cleaning DB"

	for new_user in updates.find():
	
		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})

		try:
			print "checking: " + user['username']
		except Exception as e:
			print e
		
		if user is None:
			old_user = old_users.find_one({"_id":user_id})
			user = users.find_one({"username":old_user['username']})		

		if check_banned(user['username']):
			continue
	
		try:	
			if profile_on_blockchain(user["username"],user["profile"]):
				print "cleaning: " + user["username"]
				updates.remove(new_user)
			else:
				print "Problem: " + user['username']
		except:
			pass

	for new_user in transfer.find():
		
		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})
		
		if user is None:
			old_user = old_users.find_one({"_id":user_id})
			user = users.find_one({"username":old_user['username']}) 		

		if check_banned(user['username']):
			continue

		#print "checking: " + user['username']
		if sweep_btc(new_user):
			print "Sweep BTC"
		else:
			if profile_on_blockchain(user["username"],user["profile"]):
				print "cleaning: " + user["username"]
				#transfer.remove(new_user)

	for new_user in registrations.find():
	
		user_id = new_user['user_id']
		user = users.find_one({"_id":user_id})

		if check_banned(user['username']):
			continue		

		try:
			if profile_on_blockchain(user["username"],user["profile"]):
				print "cleaning: " + user["username"]
				registrations.remove(new_user)

		except:
			pass

	print "----------"

#-----------------------------------
if __name__ == '__main__':

	#check_users()
	check_transfer()
	update_users()
	
	register_users()
	
	cleanup_db()