tweetwall/anxwall.py at master · CoreFloDev/tweetwall · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
# -*- coding: utf-8 -*-
import re
from twitter import *
from dateutil import parser
import pytz
from termcolor import cprint, colored
import os
import time
import sys
import config
import nmap
import urllib2
import twitter

####Functions
#Clear console screen
def cls():
    os.system(['clear','cls'][os.name == 'nt'])

def header(nbrOfCon):
    cprint(".____/\       ", "magenta")
    cprint("|    )/  _____    ____   ____   ____ ___  ___ ____  ", "magenta")
    cprint("|    |   \__  \  /    \ /    \_/ __ \\  \/  // __ \ ", "magenta")
    cprint("|    |___ / __ \|   |  \   |  \  ___/ >    <\  ___/ ", "magenta")
    cprint("|_______ (____  /___|  /___|  /\___  >__/\_ \\___  >", "magenta")
    cprint("        \/    \/     \/     \/     \/      \/    \/ ", "magenta")
    print("Bienvenue sur le mur de l'annexe - "+ str(nbrOfCon) +" machines connectées au WIFI")
    print("Hashtag " + colored(config.HASHTAG, "cyan") + " pour apparaitre sur le mur")

def clean(nbrOfCon):
    cls()
    header(nbrOfCon)

def getNumOfCon():
    nm.scan(hosts='192.168.1.0/24', arguments='-n -sP -T4')
    hosts_list = [(x) for x in nm.all_hosts()]
    return len(hosts_list)

def loadTweets(twit,nm):
    clean(getNumOfCon())

    tweets = None
    error = None

    try:
        tweets = twit.search.tweets(q=config.HASHTAG)
    except urllib2.URLError:
        error = "Oups ! Je n'arrive plus à me connecter au réseau"
        pass
    except twitter.api.TwitterHTTPError:
        error = "Oups ! Je n'arrive plus à me connecter à Twitter"
        pass
    except:
        error = "Oups ! Je n'arrive plus à récupérer les tweets"
        pass

    if error is not None:
       print "\r\n"+colored(error, "red")

    if tweets is not None:
        nbrofTweet = min(config.MAX_TWEETS, len(tweets['statuses']))
        counter = nbrofTweet
        for tweet in tweets['statuses'][:nbrofTweet]:
            loadTweet(tweet)
            counter -= 1
            if counter != 0:
                print
            sys.stdout.flush()

def loadTweet(tweet):
    print(" ")
    date_utc = parser.parse(tweet['created_at']).replace(tzinfo=pytz.utc)
    local_tz = pytz.timezone("Europe/Paris")
    date = date_utc.astimezone(local_tz)
    text = tweet['text']

    # Strip URLs
    text = re.sub(r'https?:\/\/[a-zA-Z0-9\.\/\+\?]+', '', text).strip()

    print colored(date.strftime("[%H:%M:%S]"), "cyan"),
    print u" " + colored(u"\n".join(text.split("\n")[:3]), "yellow"),
    print u" — " + colored(tweet['user']['name'], "red"),

# Authentication based on https://pypi.python.org/pypi/twitter
twit = Twitter(
    auth=OAuth(config.OAUTH_TOKEN, config.OAUTH_SECRET,
               config.CONSUMER_KEY, config.CONSUMER_SECRET))
nm = nmap.PortScanner() # creates an'instance of nmap.PortScanner

#TODO Do this with a scheduler
while True:
    loadTweets(twit,nm)
    time.sleep(config.REFRESH_TIME)