voussoir
/
timesearch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename these arguments to indicate they are db rows.

master
Ethan Dalool 2020-01-05 19:49:27 -08:00
parent 927d5c57a0
commit 7dc240e905
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
timesearch/offline_reading.py
View File

@ -364,22 +364,22 @@ def trees_from_database(database, specific_submission=None):
if not found_some_posts: if not found_some_posts:
raise Exception('Found no submissions!') raise Exception('Found no submissions!')
def tree_from_submission(submission, commentpool): def tree_from_submission(submission_dbrow, comments_dbrows):
''' '''
Given the sqlite data for a submission and all of its comments, Given the sqlite data for a submission and all of its comments,
return a tree with the submission id as the root return a tree with the submission id as the root
''' '''
submission = DBEntry(submission) submission = DBEntry(submission_dbrow)
commentpool = [DBEntry(c) for c in commentpool] comments = [DBEntry(c) for c in comments_dbrows]
commentpool.sort(key=lambda x: x.created) comments.sort(key=lambda x: x.created)
print('Building tree for %s (%d comments)' % (submission.idstr, len(commentpool))) print('Building tree for %s (%d comments)' % (submission.idstr, len(comments)))
# Thanks Martin Schmidt for the algorithm # Thanks Martin Schmidt for the algorithm
# http://stackoverflow.com/a/29942118/5430534 # http://stackoverflow.com/a/29942118/5430534
tree = TreeNode(identifier=submission.idstr, data=submission) tree = TreeNode(identifier=submission.idstr, data=submission)
node_map = {} node_map = {}
for comment in commentpool: for comment in comments:
# Ensure this comment is in a node of its own # Ensure this comment is in a node of its own
this_node = node_map.get(comment.idstr, None) this_node = node_map.get(comment.idstr, None)
if this_node: if this_node: