Commit 492c02fb authored by Francesco Poldi's avatar Francesco Poldi

Cleaned up code

parent 8b9a39e2
...@@ -7,21 +7,19 @@ from .storage import db, elasticsearch, write, panda ...@@ -7,21 +7,19 @@ from .storage import db, elasticsearch, write, panda
import logging as logme import logging as logme
follow_object = {} follow_object = []
tweets_object = [] tweets_object = []
user_object = [] user_object = []
author_list = {''} author_list = {''}
author_list.pop() author_list.pop()
_follow_list = [] __follow_object = {}
def clean_follow_list(): def clean_follow_list():
logme.debug(__name__+':clean_follow_list') logme.debug(__name__+':clean_follow_list')
global _follow_list global __follow_object
global follow_object __follow_object = {}
_follow_list = []
follow_object = {}
def datecheck(datestamp, config): def datecheck(datestamp, config):
logme.debug(__name__+':datecheck') logme.debug(__name__+':datecheck')
...@@ -84,12 +82,6 @@ def _output(obj, output, config, **extra): ...@@ -84,12 +82,6 @@ def _output(obj, output, config, **extra):
if config.Pandas and obj.__class__.__name__ == "user": if config.Pandas and obj.__class__.__name__ == "user":
logme.debug(__name__+':_output:Pandas+user') logme.debug(__name__+':_output:Pandas+user')
panda.update(obj, config) panda.update(obj, config)
if extra.get("follow_list"):
logme.debug(__name__+':_output:follow_list')
follow_object.username = config.Username
follow_object.action = config.Following*"following" + config.Followers*"followers"
follow_object.users = _follow_list
panda.update(follow_object, config.Essid)
if config.Elasticsearch: if config.Elasticsearch:
logme.debug(__name__+':_output:Elasticsearch') logme.debug(__name__+':_output:Elasticsearch')
print("", end=".", flush=True) print("", end=".", flush=True)
...@@ -186,6 +178,7 @@ async def Users(u, config, conn): ...@@ -186,6 +178,7 @@ async def Users(u, config, conn):
async def Username(username, config, conn): async def Username(username, config, conn):
logme.debug(__name__+':Username') logme.debug(__name__+':Username')
global __follow_object
global follow_object global follow_object
follow_var = config.Following*"following" + config.Followers*"followers" follow_var = config.Following*"following" + config.Followers*"followers"
...@@ -197,14 +190,17 @@ async def Username(username, config, conn): ...@@ -197,14 +190,17 @@ async def Username(username, config, conn):
logme.debug(__name__+':Username:Elasticsearch') logme.debug(__name__+':Username:Elasticsearch')
elasticsearch.Follow(username, config) elasticsearch.Follow(username, config)
if config.Store_object or config.Pandas: if config.Store_object:
follow_object.append(username)
if config.Pandas:
logme.debug(__name__+':Username:object+pandas') logme.debug(__name__+':Username:object+pandas')
try: try:
_ = follow_object[config.Username][follow_var] _ = __follow_object[config.Username][follow_var]
except KeyError: except KeyError:
follow_object.update({config.Username: {follow_var: []}}) __follow_object.update({config.Username: {follow_var: []}})
follow_object[config.Username][follow_var].append(username) __follow_object[config.Username][follow_var].append(username)
if config.Pandas_au: if config.Pandas_au:
logme.debug(__name__+':Username:object+pandas+au') logme.debug(__name__+':Username:object+pandas+au')
panda.update(follow_object[config.Username], config) panda.update(__follow_object[config.Username], config)
_output(username, username, config, follow_list=_follow_list) _output(username, username, config)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment