import pickle
from VK_Tools import func
from VK_Tools import VK_connecter
from Customers_Data_Tools.words_cloud import words_cloud_user, words_cloud_text

test_id = 3186751
frends = func.get_friends_ids(test_id)
grops = func.get_groups(test_id)

# walls extraction \ [(uid, {post_id: text)}]
if True:
    wolls = []
    for i, frend in enumerate(frends):
        woll = VK_connecter.get_wall(frend)
        wolls.append(woll)
        print('num {0} from {1} is redy'.format(i, len(frends)))

    with open('frends.pkl', 'wb') as f:
        pickle.dump(wolls, f)

if True:
    wolls = []
    for i, grup in enumerate(grops):
        ngrup = int(grup) * -1
        woll = VK_connecter.get_wall(ngrup)
        wolls.append(woll)
        print('num {0} from {1} iz redy'.format(i, len(grops)))
    with open('grops.pkl', 'wb') as f:
        pickle.dump(wolls, f)

# load an info from the dumps
with open('frends.pkl', 'rb') as f:
    frends_wolls = pickle.load(f)
with open('grops.pkl', 'rb') as f:
    grops_wolls = pickle.load(f)

# extract a likers from the walls
if True:
    answs = {}
    for i, elem in enumerate(frends_wolls):
        frends_dct = {}
        uid = elem[0]
        postes = [k for k, v in elem[1].items()]
        frends_dct[uid] = postes
        ans = VK_connecter.likers_execute(frends_dct)
        print(ans)
        try:
            answs.update(ans)
        except TypeError:
            pass

    with open('frens_lakers.pkl', 'wb') as f:
        pickle.dump(answs, f)

if True:
    answs = {}
    for i, elem in enumerate(grops_wolls):
        groops_dct = {}
        uid = elem[0]
        gid = int(uid) * -1
        postes = [k for k, v in elem[1].items()]
        groops_dct[uid] = postes
        ans = VK_connecter.likers_execute(groops_dct)
        print(ans)
        try:
            answs.update(ans)
        except TypeError:
            pass

    with open('grops_lakers.pkl', 'wb') as f:
        pickle.dump(answs, f)

# upload from dumps
with open('frens_lakers.pkl', 'rb') as f:
    friends_likers = pickle.load(f)
with open('grops_lakers.pkl', 'rb') as f:
    groups_likers = pickle.load(f)

# find the posts which were liked by test_id
if False:
    friends_posts_liked = []
    for k, v in friends_likers.items():
        if test_id in v:
            friends_posts_liked.append(k)
    with open('friends_posts_liked.pkl', 'wb') as f:
        pickle.dump(friends_posts_liked, f)

    groups_posts_liked = []
    for k, v in groups_likers.items():
        if test_id in v:
            groups_posts_liked.append(k)
    with open('groups_posts_liked.pkl', 'wb') as f:
        pickle.dump(groups_posts_liked, f)

# load from da dumps
with open('friends_posts_liked.pkl', 'rb') as f:
    friends_posts_liked = pickle.load(f)

with open('groups_posts_liked.pkl', 'rb') as f:
    groups_posts_liked = pickle.load(f)

# extract liked texts
friends_liked_texts_raw = []
for elem in friends_posts_liked:
    owner, postid = elem[0], elem[1]
    for subelem in frends_wolls:
        if subelem[0] == owner:
            friends_liked_texts_raw.append(subelem)

groups_liked_texts_raw = []
for elem in groups_posts_liked:
    owner, postid = elem[0], elem[1]
    for subelem in grops_wolls:
        if subelem[0] == owner:
            groups_liked_texts_raw.append(subelem)

# fuse texts in da strings objects
friends_l_texts, groups_l_texts = [], []
for elem in friends_liked_texts_raw:
    content = elem[1]
    text = ' '.join([v for k, v in content.items()])
    friends_l_texts.append(text)

for elem in groups_liked_texts_raw:
    content = elem[1]
    text = ' '.join([v for k, v in content.items()])
    groups_l_texts.append(text)

fav_texts = friends_l_texts + groups_l_texts
# please find entire text of liked posts in da fav_texts
fav_texts_strght = ' '.join(fav_texts)
wc = words_cloud_user(test_id)
wclks = words_cloud_text(fav_texts_strght, verbose=True)
print('my favorite words: ')
print(wc)
print('favorite words of liked posts: ')
print(wclks)
f = open('result.txt', 'w')
f.write('my favorite words: \n')
f.write(' '.join(wc))
f.write('\n')
f.write('fav words of liked: \n')
f.write(' '.join(wclks))
f.write('\n')
f.close()