import requests
from bs4 import BeautifulSoup


def parsepost(post):
    pic_num = 0
    doc_num = 0
    if_poll, if_longread, text_link = None, None, None
    vid_num = 0
    links = []
    views = post['response']['items'][0]['views']['count']
    items = post['response']['items'][0]
    for i in range(len(items['attachments'])):
        if items['attachments'][i]['type'] == 'photo':
            pic_num += 1
        if items['attachments'][i]['type'] == 'video':
            vid_num += 1
        if items['attachments'][i]['type'] == 'link' and 'm.vk.com/@' in items['attachments'][i]['link']['url']:
            if_longread = True
            text_link = items['attachments'][i]['link']['url']
            html = requests.get(text_link).text
            soup = BeautifulSoup(html, 'html.parser').find()
            for k in soup.find_all('a', title=True):
                links.append(k['title'])
        if items['attachments'][i]['type'] == 'poll':
            if_poll = True
        if items['attachments'][i]['type'] == 'doc':
            doc_num += 1
    return{"pic_num" : pic_num, "doc_num" : doc_num, "vid_num" : vid_num, "if_poll" : if_poll, "if_longread" : if_longread, "links" : links, 'views' : views}

def parsedoc():
    pass