tt_user.py

import twint
from datetime import datetime
import time
import os

data = datetime.now()
timestr = time.strftime("%Y%m%d")
buscas = []

print('-'*70)
print('-Script para raspagem de dados do Twitter-\n'
      'Defina os parâmetros para sua busca.\n'
      'A saída gera um relatório da busca e um arquivo CSV com os dados.\n')
print('-'*70)

quantidadeDeBuscas = input('Digite o número de buscas a ser realizada : ')
parametro = input('Busca por termo (1) ou por usuário (2)? ')
c = twint.Config()  # configurar os parâmatros de busca do twint
for termo in range(int(quantidadeDeBuscas)):

    if parametro == '1':
        busca = input('Digite o termo da busca: ')
        buscas.append(busca)
        c.Search = "'"+busca+"'"
        c.Username = None
        nome = busca+'_'+timestr
        c.Output = os.path.join('DATA', nome)  # pasta de saída
        if not os.path.exists(c.Output):  # se a pasta não existe, cria a pasta DATA
            os.makedirs(c.Output)

    elif parametro == '2':
        busca = input('Digite o nome do usuário: ')
        buscas.append(busca)
        c.Search = None
        c.Username = busca
        nome = 'user_'+c.Username+'_'+timestr
        c.Output = os.path.join('DATA', nome)
        if not os.path.exists(c.Output):
            os.makedirs(c.Output)
    else:
        print('Reinicie o script e escolha a opção da busca.')
        exit()

lang = input('Digite o idioma da busca (pt, en, es, fr, todos): ')
if lang == 'todos':
    c.Lang = None
else:
    c.Lang = lang

período = input('Digite a data e hora de início da busca (AAAA-MM-DD HH:MM:SS)'
                ' ou deixe em branco para ignorar esse parâmetro: ')
if período == '':
    c.Since = None
else:
    c.Since = período

períodoFinal = input('Digite a data e hora de final da busca (AAAA-MM-DD HH:MM:SS)'
                     ' ou deixe em branco para ignorar esse parâmetro: ')
if períodoFinal == '':
    c.Until = None
else:
    c.Until = períodoFinal

c.Store_csv = True  # define o arquivo final como csv
c.Tabs = True  # define que a separação das colunas do CSV serão através de tabulação
c.Hide_output = True  # esconde a raspagem em tempo real

# imprime o resumo dos parâmetros
print('-'*70)
for item in buscas:
    print(f'Iniciando busca para : {item}')
    if parametro == '1':
        c.Search = "'"+item+"'"
        c.Username = None
        nome = item+'_'+timestr
        c.Output = os.path.join('DATA', nome)  # pasta de saída
        # se a pasta não existe, cria a pasta DATA
        if not os.path.exists(c.Output):
            os.makedirs(c.Output)
        elif parametro == '2':
            c.Search = None
            c.Username = item
            nome = 'user_'+c.Username+'_'+timestr
            c.Output = os.path.join('DATA', nome)
    # criação do relatório com os dados da busca:
    reportPath = os.path.join(c.Output, 'relatório')
    relatório = open(f'{reportPath}_{nome}.txt', 'w')
    relatório.write(
        '-Raspagem do Twitter-\n'
        f'Termo da busca: {c.Search};\n'
        f'Usuário buscado: {c.Username};\n'
        f'Data e hora da busca: {data}; \n'
        f'Idioma da busca: {c.Lang};\n'
        f'Data do início da busca: {c.Since};\n'
        f'Data do final da busca: {c.Until};\n'
        f'Nome da pasta: {nome}.')
    relatório.close
    twint.run.Search(c)