Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed mongodb queries to return genesis block and last voted block

Browse Source
This commit is contained in:
Rodolphe Marques 2016-12-22 14:30:19 +01:00
parent 9f738cd58d
commit a0952df9fb
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bigchaindb/backend/mongodb/query.py
View File

@ -133,8 +133,9 @@ def write_vote(conn, vote):
@register_query(MongoDBConnection) @register_query(MongoDBConnection)
def get_genesis_block(conn): def get_genesis_block(conn):
return conn.db['bigchain'].find_one({'block.transactions.0.operation' == return conn.db['bigchain'].find_one({
'GENESIS'}) 'block.transactions.0.operation': 'GENESIS'
})
@register_query(MongoDBConnection) @register_query(MongoDBConnection)
@ -142,7 +143,10 @@ def get_last_voted_block(conn, node_pubkey):
last_voted = conn.db['votes']\ last_voted = conn.db['votes']\
.find({'node_pubkey': node_pubkey}, .find({'node_pubkey': node_pubkey},
sort=[('vote.timestamp', -1)]) sort=[('vote.timestamp', -1)])
if not last_voted:
# pymongo seems to return a cursor even if there are no results
# so we actually need to check the count
if last_voted.count() == 0:
return get_genesis_block(conn) return get_genesis_block(conn)
mapping = {v['vote']['previous_block']: v['vote']['voting_for_block'] mapping = {v['vote']['previous_block']: v['vote']['voting_for_block']