Talho/vkproxy.py at master · radioanonymous/Talho · 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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/usr/bin/env python
# coding: utf-8

import sys, os
import time
import urllib, urllib2
import re
import traceback
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
from SocketServer import ThreadingMixIn
import socket
from vkproxy_settings import vk_acc, icecast_admin

opener = urllib2.build_opener(urllib2.HTTPCookieProcessor())

class IcecastAdminOpener(urllib.FancyURLopener):
    def prompt_user_passwd(self, host, realm):
        return icecast_admin

def set_tag(args):
    if not args:
        return
    try:
        song_name = " ".join(args)[:100]
        metadata_opener = IcecastAdminOpener()
        update_query = urllib.urlencode({ "mount" : "/radio", "mode" : "updinfo", "song" : song_name })
        metadata_opener.open("http://127.0.0.1:8000/admin/metadata?" + update_query).read()
    except:
        pass

class VkHandler(BaseHTTPRequestHandler):

    def do_GET(self):
        global opener
        dl_name = os.tmpnam()
        pathsplit = self.path[1:].split("/")
        songtitle = pathsplit[0]
        if len(pathsplit) > 1 and pathsplit[1].isdigit():
            songshift = int(pathsplit[1])
        else:
            songshift = 0

        try:
            #result = handle.read().decode("cp1251")
            req = urllib2.Request("http://vkontakte.ru/search?c%%5Bq%%5D=%s&c%%5Bsection%%5D=audio" % songtitle)

            handle = opener.open(req)
            result = handle.read()
            handle.close()
            result = result.decode("cp1251", "ignore")

            match = re.findall(ur"value=\"(http:[^,]+),", result)
            if match:
                link = match[songshift]
                mp3_handle = urllib2.urlopen(link)
	        self.send_response(200)
                self.send_header('Content-Type', mp3_handle.headers['Content-Type'])
                self.send_header('Content-Length', mp3_handle.headers['Content-Length'])
                self.end_headers()
	#tmpfile = open(dl_name, "wb")

                print "Transfer started"
                #if self.client_address[0] == "127.0.0.1":
                #    set_tag(songtitle.split("+"))
                while True:
                    chunk = mp3_handle.read(128 * 1024)
                    if not chunk:
                        break
                    self.wfile.write(chunk)
		    #tmpfile.write(chunk)

                print "Transfer finished"
		#tmpfile.close()
                mp3_handle.close()
		#tmpfile = open(dl_name, "rb")
                #while True:
                #    chunk = tmpfile.read(16 * 1024)
                #    if not chunk:
                #        break
                #    self.wfile.write(chunk)
		#tmpfile.close()
		self.wfile.close()
		#os.unlink(dl_name)
                return

	except socket.error, msg:
            print str("".join(traceback.format_exception(*sys.exc_info())))
	    #os.unlink(dl_name)
	    return
        except:
            print str("".join(traceback.format_exception(*sys.exc_info())))
            print '--------------------------------------------------------'
	    self.send_response(200)
	    return

        self.send_response(404)
        return

    def do_HEAD(self):
        self.send_response(200)

class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
    pass

url = "http://vkontakte.ru/login.php?email=%s&pass=%s" % vk_acc
req = urllib2.Request(url)
handle = opener.open(req)
socket.setdefaulttimeout(30)

server = ThreadedHTTPServer(('0.0.0.0', 8080), VkHandler)
server.serve_forever()