diff options
Diffstat (limited to 'dumpfm/dumpfm.py')
| -rw-r--r-- | dumpfm/dumpfm.py | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/dumpfm/dumpfm.py b/dumpfm/dumpfm.py index a6138b4..5bffdbc 100644 --- a/dumpfm/dumpfm.py +++ b/dumpfm/dumpfm.py @@ -156,5 +156,21 @@ engine = create_engine('postgresql://{}@{}/{}'.format( Session = sessionmaker(bind=engine) session = Session() -message_list = session.query(Message).filter_by(is_image=True).all() -print len(message_list) +message_list = session.query(Message).filter_by(is_image=True) +count = 0 +import sys +import re +f = open("IDLIST2", "w") +exp = re.compile(r'.*asdf.us/im/.*') +exp2 = re.compile(r'.*favor.*', re.IGNORECASE) +for message in message_list: + if exp.match(message.content): + session2 = Session() + favorites_count = 0 + tags = session2.query(Tag).filter_by(message_id=message.message_id) + for tag in tags: + if exp2.match(tag.tag): + favorites_count += 1 + session2.close() + f.write("%s\t%s\t%s\n" % ( message.message_id, message.content.encode('utf8'), favorites_count)) + |