facebook_info_scraping.py

#!usr/bin/env python3
# -*- coding: utf-8 -*-
#
# @author Nastasia Vanderperren
#
# get the info of a fb page, group or account
# returns a json file
#

from argparse import ArgumentParser
from facebook_scraper import get_group_info, get_profile, get_page_info
from json import dumps
from datetime import date

def write_info(info, account):
    today = date.today().strftime("%Y%m%d")
    json_file = dumps(info, indent=2, ensure_ascii=False)
    with open('{}_{}_info.json'.format(today, account), 'w') as output_file:
        output_file.write(json_file)
        output_file.close()


def get_fb_info(args):
    account = args.account
    cookies = args.cookies if args.cookies else None
    friends = args.friends
    type = args.type
    info = None

    if type == 'group':
        info = get_group_info(account, cookies=cookies, friends=friends) 
    if type == 'account':
        info = get_profile(account, cookies=cookies)
    if type == 'page':
        info = get_page_info(account, cookies=cookies)

    write_info(info, account)


if __name__ == '__main__':
    parser = ArgumentParser()
    parser.add_argument('account', help="name of the account", type=str)
    parser.add_argument('--cookies', help="cookie file for getting data of a private account", required=False)
    parser.add_argument('--friends', help="extract friends of accounts", action='store_true', required=False)
    parser.add_argument('--type', help="which type of account would you like to scrape", choices=['account', 'group', 'page'], required=True)
    args = parser.parse_args()
    get_fb_info(args)