mirror of https://github.com/CIRCL/AIL-framework
48 lines
1.3 KiB
Python
Executable File
48 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*-coding:UTF-8 -*
|
|
"""
|
|
The JSON Receiver Module
|
|
================
|
|
|
|
Recieve Json Items (example: Twitter feeder)
|
|
|
|
"""
|
|
import os
|
|
import json
|
|
import sys
|
|
import datetime
|
|
|
|
sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib'))
|
|
import item_basic
|
|
import Username
|
|
|
|
sys.path.append(os.path.join(os.environ['AIL_BIN'], 'import', 'ail_json_importer'))
|
|
from Default_json import Default_json
|
|
|
|
class Ail_feeder_twitter(Default_json):
|
|
"""Twitter Feeder functions"""
|
|
|
|
def __init__(self, name, json_item):
|
|
super().__init__(name, json_item)
|
|
|
|
def get_feeder_name(self):
|
|
return 'twitter'
|
|
|
|
# define item id
|
|
def get_item_id(self):
|
|
# use twitter timestamp ?
|
|
item_date = datetime.date.today().strftime("%Y/%m/%d")
|
|
item_id = str(self.json_item['meta']['twitter:tweet_id'])
|
|
return os.path.join('twitter', item_date, item_id) + '.gz'
|
|
|
|
def process_json_meta(self, process, item_id):
|
|
'''
|
|
Process JSON meta filed.
|
|
'''
|
|
twitter_id = str(self.json_item['meta']['twitter:tweet_id'])
|
|
item_basic.add_map_obj_id_item_id(twitter_id, item_id, 'twitter_id')
|
|
username = str(self.json_item['meta']['twitter:id'])
|
|
item_date = item_basic.get_item_date(item_id)
|
|
Username.save_item_correlation('twitter', username, item_id, item_date)
|
|
return None
|