UDPServer.py

## TODO: Accept command line args

#!/usr/bin/python

import sys
import getopt
import argparse
import threading
import Queue
import socket
import time
import logging
from bs4 import BeautifulSoup
import urllib2

UDP_IP_S = "127.0.0.1"
UDP_PORT_S = 13337
UDP_IP_R = "127.0.0.1"
UDP_PORT_R = 13338
MESSAGE = "TEMP"
bots = 0
exit = 0
cmd = 1
url = "http://176.31.191.50/index.html"

def main(argv):
    global url

    parser = argparse.ArgumentParser(usage='Break all the things')
    parser.add_argument('URL', help='URL to take config from')
    cliserv = parser.add_mutually_exclusive_group(required=True)
    cliserv.add_argument('-s', '--server', help='Run the script in server mode', action='store_true')
    cliserv.add_argument('-c', '--client', help='Run the script in client mode', action='store_true')

    args = parser.parse_args()
    url = args.URL
    return

def sender(): # sends until something changes exit to 1
    global exit
    print 'Sending...'
    ## TODO: Make this get a command from the parser
    while exit == 0:
        sockS = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        sockS.sendto(MESSAGE, (UDP_IP_S, UDP_PORT_S))
        time.sleep(1)
    return

def listener():
    global bots
    global exit
    while bots <= 3:
        print 'Listening...'
        sockR = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        sockR.bind((UDP_IP_R, UDP_PORT_R))

        data, addr = sockR.recvfrom(1024) # buffer size is 1024 bytes
        sockR.close()
        print "New Bot:", data
        workers += 1
    exit = 1
    return

def checker(): ## TODO: Accept a URL as input maybe?
    while exit == 0:
        print 'Parsing...'
        global cmd
        global url
        response = urllib2.urlopen(url)
        html = response.read()
        soup = BeautifulSoup(html)

        cmd = len(soup.a.string) # gets the length of the first name
        print 'Parsing result is', cmd
        time.sleep(10)
    return

def threads():
    s = threading.Thread(name='Sender', target=sender)
    l = threading.Thread(name='Listener', target=listener)
    c = threading.Thread(name='Checker', target=checker)

    c.start()
    time.sleep(2) # give the parser a hot second
    s.start()
    l.start()

if __name__ == "__main__":
    main(sys.argv[1:])