# Path: database.py
# def txn_required(func):
#     @functools.wraps(func)
#     def _func(*args, **kwargs):
#         txn = current_txn()
#         if txn:
#             return func(*args, **kwargs)
#         else:
#             return run_in_transaction(func, txn_name=func.__name__, *args, **kwargs)
#     return _func
# 
# class TxnAbort(Exception):
#     def __init__(self, callback=None, *args, **kwargs):
#         self.__callback = callback
#         self.__args = args
#         self.__kwargs = kwargs
#         
#     @property
#     def callback(self):
#         if self.__callback:
#             return lambda: self.__callback(*self.__args, **self.__kwargs)
#         else:
#             return None
#             
#     def __call__(self):
#         cb = self.callback
#         if cb: return cb()
# 
# @contextmanager
# def Transaction(name="*UNKNOWN*", flags=DEFAULT_TXN_FLAGS):
#         txn = begin_txn(flags)
#         txn.set_name(name)
#         txn_log.debug("TXN BEGIN %s", repr(txn))
#         try:
#             yield txn
#         except Exception as e:
#             txn_log.debug("TXN ABORT %s", repr(txn)) 
#             abort_txn()
#             raise
#         txn_log.debug("TXN COMMIT %s", repr(txn))
#         commit_txn()


import eventlet
import genesisblock
import msgs
import jserialize as js
import bserialize as bs
import logging
import settings
import database
import transactions
import struct
    import sys
    import time
        import debug
from utils import *
from database import txn_required, TxnAbort, Transaction
    
def get_bestblock(txn=None, raw=False):
    h = state.get("bestblock")
    if raw:
        return h
    return Block.get_by_hash(h)
 
@txn_required
def set_bestblock(blk, check=True):
    if check:
        best = get_bestblock()
        if blk.totaldiff <= best.totaldiff:
            log.info("%s(%d) is not better then %s(%d)", h2h(blk.hash), blk.number, h2h(best.hash), best.number)
            return False
        if best.hash != blk.prev:
            reorganize(best, blk)
    state.put("bestblock", blk.hash)
    return True

@txn_required
def reorganize(old, new):
    log.info("reorginizing blockchain:")
    split, old_blocks, new_blocks = find_split(old, new)
    log.info("REORG: split:%s(%d)", h2h(split.hash), split.number)
    for blk in old_blocks:
        blk.revert()
    for blk in reversed(new_blocks):
        try:
            blk.confirm()
        except InvalidBlock:

# Path: database.py
# def txn_required(func):
#     @functools.wraps(func)
#     def _func(*args, **kwargs):
#         txn = current_txn()
#         if txn:
#             return func(*args, **kwargs)
#         else:
#             return run_in_transaction(func, txn_name=func.__name__, *args, **kwargs)
#     return _func
# 
# class TxnAbort(Exception):
#     def __init__(self, callback=None, *args, **kwargs):
#         self.__callback = callback
#         self.__args = args
#         self.__kwargs = kwargs
#         
#     @property
#     def callback(self):
#         if self.__callback:
#             return lambda: self.__callback(*self.__args, **self.__kwargs)
#         else:
#             return None
#             
#     def __call__(self):
#         cb = self.callback
#         if cb: return cb()
# 
# @contextmanager
# def Transaction(name="*UNKNOWN*", flags=DEFAULT_TXN_FLAGS):
#         txn = begin_txn(flags)
#         txn.set_name(name)
#         txn_log.debug("TXN BEGIN %s", repr(txn))
#         try:
#             yield txn
#         except Exception as e:
#             txn_log.debug("TXN ABORT %s", repr(txn)) 
#             abort_txn()
#             raise
#         txn_log.debug("TXN COMMIT %s", repr(txn))
#         commit_txn()


# Path: blockchain.py
import eventlet
import genesisblock
import msgs
import jserialize as js
import bserialize as bs
import logging
import settings
import database
import transactions
import struct
    import sys
    import time
        import debug
from utils import *
from database import txn_required, TxnAbort, Transaction

    
def get_bestblock(txn=None, raw=False):
    h = state.get("bestblock")
    if raw:
        return h
    return Block.get_by_hash(h)
 
@txn_required
def set_bestblock(blk, check=True):
    if check:
        best = get_bestblock()
        if blk.totaldiff <= best.totaldiff:
            log.info("%s(%d) is not better then %s(%d)", h2h(blk.hash), blk.number, h2h(best.hash), best.number)
            return False
        if best.hash != blk.prev:
            reorganize(best, blk)
    state.put("bestblock", blk.hash)
    return True

@txn_required
def reorganize(old, new):
    log.info("reorginizing blockchain:")
    split, old_blocks, new_blocks = find_split(old, new)
    log.info("REORG: split:%s(%d)", h2h(split.hash), split.number)
    for blk in old_blocks:
        blk.revert()
    for blk in reversed(new_blocks):
        try:
            blk.confirm()
        except InvalidBlock:

            raise TxnAbort(blk.invalidate)