#!/usr/bin/python
import sys
import os
import os.path
import re
import urllib2
import simplejson
from optparse import OptionParser
class YTMND:
def __init__ (self):
self.media_only = False
self.no_web_audio = False
self.json = False
def fetch_user(self, user):
if user == "":
print("expecting one ytmnd name, got "+str(sys.argv))
return
# Get the url of the sound and foreground
ytmnd_name = user
ytmnd_html = urllib2.urlopen("http://ytmnd.com/users/" + ytmnd_name + "/sites").readlines()
domains = []
for line in ytmnd_html:
if 'profile_link' in line:
expr = r"site_link\" href=\"http://(\S+).ytmn(d|sfw)?.com\""
domain = re.search(expr,line).group(1)
domains.append(domain)
print ">> found %d domains" % len( domains )
os.system("mkdir -p %s" % user)
os.chdir(user)
if not self.no_web_audio:
self.copy_ytmnd_js()
for domain in domains:
ytmnd.fetch_ytmnd( domain )
os.chdir("..")
def fetch_ytmnd(self, domain):
if domain == "":
print("expecting one ytmnd name, got "+str(sys.argv))
return
print "fetching %s" % domain
# Get the url of the sound and foreground
ytmnd_name = domain
ytmnd_html = urllib2.urlopen("http://" + domain + ".ytmnd.com").read()
expr = r"ytmnd.site_id = (\d+);"
ytmnd_id = re.search(expr,ytmnd_html).group(1)
ytmnd_info = simplejson.load(urllib2.urlopen("http://" + domain + ".ytmnd.com/info/" + ytmnd_id + "/json"))
if ytmnd.json:
print simplejson.dumps(ytmnd_info, sort_keys=True, indent=4 * ' ')
# ytmnd.write_json(ytmnd_info)
else:
ytmnd.fetch_media(ytmnd_info)
if not ytmnd.media_only:
ytmnd.write_index(ytmnd_info)
def fetch_media(self, ytmnd_info):
# Assign full url names for the sound and foreground
domain = ytmnd_info['site']['domain']
original_gif = ytmnd_info['site']['foreground']['url']
original_wav = ytmnd_info['site']['sound']['url']
if 'alternates' in ytmnd_info['site']['sound']:
key, value = ytmnd_info['site']['sound']['alternates'].popitem()
original_wav = value['file_url']
# download files
os.system("wget --quiet -O %s %s" % (domain + ".gif", original_gif))
os.system("wget --quiet -O %s %s" % (domain + ".mp3", original_wav))
def write_index(self, ytmnd_info):
# print simplejson.dumps(ytmnd_info)
domain = ytmnd_info['site']['domain']
bgcolor = ytmnd_info['site']['background']['color']
title = ytmnd_info['site']['description']
placement = ytmnd_info['site']['foreground']['placement']
fn = open(domain + ".html", 'w')
fn.write("\n")
fn.write("\n")
fn.write("%s\n" % title)
fn.write("\n")
fn.write("\n")
if self.no_web_audio:
fn.write("