def main():
parser = argparse.ArgumentParser(description='Label author-used IP')
parser.add_argument('-b', metavar='BOARD_NAME', help='Board name', required=True)
args = parser.parse_args()
board = args.b
client = MongoClient('localhost', 27017)
collection = client['Ptt'][board]
articles = collection.find()
for article in articles:
if 'school_data' not in article:
school_data = findip.findIP_School(article['ip'])
try:
collection.update({'_id':article['_id']}, {'$set':school_data})
except pymongo.errors.CursorNotFound:
print "No article in database"
print article['article_title'], article['ip'], school_data['school'], school_data['place']
ptt.loginPtt()
ptt.enterUserList()
client = MongoClient('localhost', 27017)
db = client['Ptt']
collect = db['Gossiping']
articles = collect.find()
for article in articles:
for message in article['messages']:
print(message)
if 'push_userid' in message and 'school' not in message:
push_userid = message['push_userid']
user_ip = ptt.getUserIP(push_userid)
school_data = findip.findIP_School(user_ip)
school_data['ip'] = user_ip
collect.update(
{
'_id': article['_id'],
'messages':{
'$elemMatch':{
'push_userid': push_userid,
'ip': {'$nin': [user_ip]}
}
}
},
{'$set': {
'messages.$.ip': user_ip,
