dotanoobs/teamspeak.py

import operator
import os
import ts3
import time
import requests
from xml.etree import ElementTree
from flask import url_for
from bs4 import BeautifulSoup

from app import app

def getTeamspeakWindow(window=605800):
    current_time = time.time()
    from models import TeamspeakData
    return TeamspeakData.query.filter(TeamspeakData.time < current_time, TeamspeakData.time > current_time-window).order_by(TeamspeakData.time).all()

def create_teamspeak_viewer():
    try:
            server = ts3.TS3Server(app.config['TS3_HOST'], app.config['TS3_PORT'])
            server.login(app.config['TS3_USERNAME'], app.config['TS3_PASSWORD'])
            server.use(1)

            serverinfo = server.send_command('serverinfo').data
            channellist = server.send_command('channellist', opts=("limits", "flags", "voice", "icon")).data
            clientlist = server.send_command('clientlist', opts=("away", "voice", "info", "icon", "groups", "country")).data
            servergrouplist = server.send_command('servergrouplist').data
            channelgrouplist = server.send_command('channelgrouplist').data

            soup = BeautifulSoup()
            div_tag = soup.new_tag('div')
            div_tag['class'] ='devmx-webviewer'
            soup.append(div_tag)

            def construct_channels(parent_tag, cid):
                    num_clients = 0
                    for channel in channellist:
                            if int(channel['pid']) == int(cid):
                                    # Construct the channel
                                    channel_tag = soup.new_tag('div')
                                    channel_tag['class'] = 'tswv-channel'
                                    # Channel image
                                    image_tag = soup.new_tag('span')
                                    image_tag['class'] = 'tswv-image tswv-image-right'
                                    if int(channel['channel_flag_password']) == 1:
                                            image_tag['class'] += ' tswv-channel-password-right'
                                    if int(channel['channel_flag_default']) == 1:
                                            image_tag['class'] += ' tswv-channel-home'
                                    if int(channel['channel_needed_talk_power']) > 0:
                                            image_tag['class'] += ' tswv-channel-moderated'
                                    if int(channel['channel_icon_id']) != 0:
                                            pass
                                    image_tag.append(' ')
                                    channel_tag.append(image_tag)
                                    # Status image
                                    status_tag = soup.new_tag('span')
                                    status_tag['class'] = 'tswv-image'
                                    if int(channel['channel_flag_password']) == 1:
                                            status_tag['class'] += ' tswv-channel-password'
                                    elif int(channel['total_clients']) == int(channel['channel_maxclients']):
                                            status_tag['class'] += ' tswv-channel-full'
                                    else:
                                            status_tag['class'] += ' tswv-channel-normal'
                                    status_tag.append(' ')
                                    channel_tag.append(status_tag)
                                    # Label
                                    label_tag = soup.new_tag('span')
                                    label_tag['class'] = 'tswv-label'
                                    label_tag.append(channel['channel_name'])
                                    channel_tag.append(label_tag)
                                    # Clients
                                    channel_tag, channel_clients = construct_clients(channel_tag, channel['cid'])
                                    # Recurse through sub-channels, collecting total number of clients as we go
                                    channel_tag, sub_clients = construct_channels(channel_tag, channel['cid'])
                                    channel_clients += sub_clients
                                    # Only show non-empty channels
                                    if channel_clients > 0:
                                            parent_tag.append(channel_tag)
                                            num_clients += channel_clients
                    return parent_tag, num_clients

            def construct_clients(parent_tag, cid):
                    num_clients = 0
                    for client in clientlist:
                            if int(client['cid']) == int(cid):
                                    # Skip ServerQuery clients
                                    if int(client['client_type']) == 1: continue
                                    num_clients += 1
                                    client_tag = soup.new_tag('div')
                                    client_tag['class'] = 'tswv-client'
                                    # Status image
                                    status_tag = soup.new_tag('span')
                                    status_tag['class'] = 'tswv-image'
                                    if int(client['client_type']) == 1:
                                            status_tag['class'] += ' tswv-client-query'
                                    elif int(client['client_away']) == 1:
                                            status_tag['class'] += " tswv-client-away"
                                    elif int(client['client_input_muted']) == 1:
                                            status_tag['class'] += " tswv-client-input-muted"
                                    elif int(client['client_output_muted']) == 1:
                                            status_tag['class'] += " tswv-client-output-muted"
                                    elif int(client['client_input_hardware']) == 0:
                                            status_tag['class'] += " tswv-client-input-muted-hardware"
                                    elif int(client['client_output_hardware']) == 0:
                                            status_tag['class'] += " tswv-client-output-muted-hardware"
                                    elif (int(client['client_flag_talking']) == 1) and (int(client['client_is_channel_commander']) == 1):
                                            status_tag['class'] += " tswv-client-channel-commander-talking"
                                    elif int(client['client_is_channel_commander']) == 1:
                                            status_tag['class'] += " tswv-client-channel-commander"
                                    elif int(client['client_flag_talking']) == 1:
                                            status_tag['class'] += " tswv-client-talking"
                                    else:
                                            status_tag['class'] += " tswv-client-normal"
                                    status_tag.append(' ')
                                    client_tag.append(status_tag)
                                    # Country image
                                    if client['client_country']:
                                            country_tag = soup.new_tag('span')
                                            country_tag['class'] = 'tswv-image tswv-image-right'
                                            country_tag['title'] = ' '.join([word.capitalize() for word in ISO3166_MAPPING[client['client_country']].split(' ')])
                                            country_tag['style'] = 'background: url("%s") center center no-repeat;' % url_for('static', filename='img/ts3_viewer/countries/%s.png' % client['client_country'].lower())
                                            country_tag.append(' ')
                                            client_tag.append(country_tag)
                                    # Server group images
                                    sgids = [int(sg) for sg in client['client_servergroups'].split(',')]
                                    servergroups = [servergroup for servergroup in servergrouplist if int(servergroup['sgid']) in sgids]
                                    servergroups.sort(key=operator.itemgetter('sortid'))
                                    for servergroup in servergroups:
                                            if not servergroup['iconid']: continue
                                            img_fname = 'img/ts3_viewer/%s.png' % servergroup['iconid']
                                            if not os.path.exists(os.path.join(app.static_folder, img_fname)):
                                                    continue
                                            image_tag = soup.new_tag('span')
                                            image_tag['class'] = 'tswv-image tswv-image-right'
                                            image_tag['title'] = servergroup['name']
                                            image_tag['style'] = 'background-image: url("%s")' % url_for('static', filename=img_fname)
                                            image_tag.append(' ')
                                            client_tag.append(image_tag)
                                    # Check if client is in a moderated channel
                                    channel = [channel for channel in channellist if int(channel['cid']) == int(client['cid'])][0]
                                    if int(channel['channel_needed_talk_power']) > 0:
                                            status_tag = soup.new_tag('span')
                                            status_tag['class'] = 'tswv-image tswv-image-right'
                                            if int(client['client_is_talker']) == 0:
                                                    status_tag['class'] += ' tswv-client-input-muted'
                                            else:
                                                    status_tag['class'] += ' tswv-client-talkpower-granted'
                                            status_tag.append(' ')
                                            client_tag.append(status_tag)
                                    # Label
                                    label_tag = soup.new_tag('span')
                                    label_tag['class'] = 'tswv-label'
                                    label_tag.append(client['client_nickname'])
                                    client_tag.append(label_tag)
                                    parent_tag.append(client_tag)
                    return parent_tag, num_clients
            div_tag, num_clients = construct_channels(div_tag, 0)
            return (soup.prettify(), num_clients)
    except Exception as inst:
            return "error: %s" % inst

def get_ISO3166_mapping():
    #data = requests.get(url_for('static', filename='country_codes.xml'))
    #xml = ElementTree.fromstring(data.text.encode('utf-8'))
    with open('app/static/country_codes.xml', mode='r') as d:
        data = d.read()
    xml = ElementTree.fromstring(data)
    d = dict()
    for entry in xml.findall('ISO_3166-1_Entry'):
        d[entry.find('ISO_3166-1_Alpha-2_Code_element').text] = entry.find('ISO_3166-1_Country_name').text
    return d

ISO3166_MAPPING = get_ISO3166_mapping()