|
|
@ -0,0 +1,230 @@ |
|
|
|
from flask import Flask |
|
|
|
from flask.ext.sqlalchemy import SQLAlchemy |
|
|
|
from flask.ext.openid import OpenID |
|
|
|
from flask.ext.cache import Cache |
|
|
|
|
|
|
|
app = Flask(__name__) |
|
|
|
app.config.from_object('config') |
|
|
|
db = SQLAlchemy(app) |
|
|
|
oid = OpenID(app) |
|
|
|
cache = Cache(app, config={'CACHE_TYPE': app.config['CACHE_TYPE']}) |
|
|
|
|
|
|
|
from app import views |
|
|
|
|
|
|
|
''' |
|
|
|
from flask import Flask, render_template |
|
|
|
from flask.ext.mongoengine import MongoEngine |
|
|
|
from flask.ext.openid import OpenID |
|
|
|
from flask.ext.cache import Cache |
|
|
|
import utils |
|
|
|
import ts3 |
|
|
|
|
|
|
|
app = Flask(__name__) |
|
|
|
app.config.from_object('config') |
|
|
|
|
|
|
|
#Setup mongo database |
|
|
|
db = MongoEngine(app) |
|
|
|
|
|
|
|
#Setup OpenID and Caching |
|
|
|
oid = OpenID(app) |
|
|
|
cache = Cache(app, config={'CACHE_TYPE': app.config['CACHE_TYPE']}) |
|
|
|
|
|
|
|
from app import views |
|
|
|
@app.route('/') |
|
|
|
def inx(): |
|
|
|
return render_template('main.html') |
|
|
|
|
|
|
|
##### INTO UTILS LATER ##### |
|
|
|
RADIANT_TEAM = 2 |
|
|
|
DIRE_TEAM = 3 |
|
|
|
RADIANT_COLOR = 'b' |
|
|
|
DIRE_COLOR = 'r' |
|
|
|
|
|
|
|
def get_hero_data(): |
|
|
|
xhr = urllib2.build_opener().open(urllib2.Request("https://api.steampowered.com/IEconDOTA2_570/GETHeroes/v0001/?key="+DOTA2_API_KEY+"&language=en_us")) |
|
|
|
data = json.load(xhr) |
|
|
|
return data |
|
|
|
|
|
|
|
@app.context_processor |
|
|
|
def utility_processor(): |
|
|
|
@cache.memoize(60*5) |
|
|
|
def ts3_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: |
|
|
|
raise NotImplementedError |
|
|
|
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 |
|
|
|
country_tag = soup.new_tag('span') |
|
|
|
country_tag['class'] = 'tswv-image tswv-image-right' |
|
|
|
country_tag['title'] = ' '.join([word.capitalize() for word in utils.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() |
|
|
|
except Exception as inst: |
|
|
|
return "error: %s" % inst |
|
|
|
def shorten_text(text, num_words=10): |
|
|
|
text = utils.fix_bad_unicode(text) |
|
|
|
space_iter = re.finditer('\s+', text) |
|
|
|
output = u'' |
|
|
|
while num_words > 0: |
|
|
|
match = space_iter.next() |
|
|
|
if not match: break |
|
|
|
output = text[:match.end()] |
|
|
|
num_words -= 1 |
|
|
|
else: |
|
|
|
output += '...' |
|
|
|
return output |
|
|
|
def num_unique_clients(teamspeak_data): |
|
|
|
unique_clients = set() |
|
|
|
for data in teamspeak_data: |
|
|
|
unique_clients.update(data.clients) |
|
|
|
return len(unique_clients) |
|
|
|
def num_unique_clients_by_country(teamspeak_data): |
|
|
|
unique_clients = {} |
|
|
|
for data in teamspeak_data: |
|
|
|
for client_id, client_data in data.clients.iteritems(): |
|
|
|
unique_clients[client_id] = (client_data['country'] or 'Unknown').lower() |
|
|
|
country = {} |
|
|
|
for client_id, country_code in unique_clients.iteritems(): |
|
|
|
country[country_code] = country.get(country_code, 0) + 1 |
|
|
|
return country |
|
|
|
def country_abbreviation_mapping(): |
|
|
|
mapping = {} |
|
|
|
for key, name in utils.ISO3166_MAPPING.iteritems(): |
|
|
|
mapping[key.lower()] = ' '.join([word.capitalize() for word in name.split(' ')]) |
|
|
|
return mapping |
|
|
|
return dict(timestamp_to_js_date=utils.timestamp_to_js_date, ts3_viewer=ts3_viewer, shorten_text=shorten_text, getTeamspeakWindow=doob.getTeamspeakWindow, |
|
|
|
num_unique_clients=num_unique_clients, |
|
|
|
num_unique_clients_by_country=num_unique_clients_by_country, |
|
|
|
country_abbreviation_mapping=country_abbreviation_mapping) |
|
|
|
|
|
|
|
|
|
|
|
''' |