main.py

# -*- coding: utf8 -*-
import os
import time
import json
from PyQt4 import QtCore, QtGui, QtSql
import main_form
import songdb
import login_dialog
import filter
import vk_auth
import vk
import threading
import utils


def refreshClicked():
    Dialog = QtGui.QDialog()
    ui = filter.Ui_Dialog()
    ui.setupUi(Dialog)
    
    #Dialog.setAttribute(QtCore.Qt.WA_DeleteOnClose)
    result = Dialog.exec_()
    
    if result == QtGui.QDialog.Accepted:
        city = ui.cityEdit.text()
        age_from = ui.spinBoxAgeFrom.text()
        age_to = ui.spinBoxAgeTo.text()
        #QtGui.QMessageBox.about(QtGui.QWidget(), "Супер прога", city)
        return city, age_from, age_to
        
    return None
    
def showLoginDialog():
    Dialog = QtGui.QDialog()
    ui = login_dialog.Ui_Dialog()
    ui.setupUi(Dialog)
    
    result = Dialog.exec_()
    
    if result == QtGui.QDialog.Accepted:
        loginText =  ui.loginEdit.text()
        passwordText = ui.passwordEdit.text()
        return loginText, passwordText

    return None
    
def peopleListClicked(index):
    row = index.row()
    global users
    rec = users.record(row)
    
    gender = ""
    
    if rec.value(3) == 1:
        gender = " пол: жен"
    else:
        gender = " пол: муж"
    
    global ui
    ui.userLabel.setText(rec.value(0) + ", " + gender + ", " + str(rec.value(4)) + " лет" )

progress = None

def enableAllControls(enable):
    return False
    
def createProgress(dialog):
    global progress
    progress = QtGui.QProgressBar(dialog)
    progress.setGeometry(200, 80, 250, 20)
    progress.move(450, 400)
    
def getUsersFromCity(api, pcity, complete):
    # 365 запросов по всем дням в году - так преодолемм ограничение не больше 1000    
    request_count = 0 # не более 3 в секунду, делаем два и ждем пару сек
    
    total_requests = 360
    done_requests = 0
    
    for m in range(1, 13): # месяцы
        for d in range(1, 4): #дни
            tot_count=10
            result = api.users.search(city=pcity, birth_day=d, birth_month=m, count=tot_count, fields='sex, bdate')
            
            #result = json.loads(result)
            #print(result)
            
            for ind in range(1, len(result)):
                age = 0
            
                if 'bdate' in result[ind]:
                    strage = result[ind]['bdate'].split('.')[-1]
                    
                    if len(strage) == 4:
                        age = 2016 - int(strage)
            
                songdb.addUser(result[ind]['uid'], result[ind]['first_name'], result[ind]['last_name'], result[ind]['sex'], pcity, age)
        
            if request_count == 2:
                time.sleep(7)
                request_count = 0
                
                global progress
                progress.setValue( int(done_requests/total_requests*100) )
            
            request_count += 1
            done_requests += 1
    
    if complete != None:
        complete()
        
def getMusicForUsers(api, userids, complete):
    codestrs = utils.getCodeStrs(userids)
    
    for str in codestrs:
        result = api.execute(code=str)
        print(result)
        
    if complete != None:
        complete()
            
def getUsersFromGroup(api, groupid):
    return None
            
def getUsersInThread(api, city, complete):
    t1 = threading.Thread(target = getUsersFromCity, args = (api, city, complete))
    t1.start()
    #t1.join()
    
def onLoadComplete():
    count = songdb.getUsersCount()
        
    ui.labelPeople.setText("Люди (" + str(count) + ")")

    progress.hide()
    
    ui.peopleFilterButton.clicked.connect(refreshClicked)
    
    users = songdb.selectUsers()
    ui.listPeople.setModel(users)
        
    ui.listPeople.clicked.connect(peopleListClicked)
    
            
if __name__ == "__main__":
    import sys
    app = QtGui.QApplication(sys.argv)
    
    if not os.path.isfile("music.db"):
        db = songdb.createDB()
    else:
        db = QtSql.QSqlDatabase.addDatabase('QSQLITE')
        db.setDatabaseName('music.db')
        
    if db is None or not db.open():
        QtGui.QMessageBox.critical(None, QtGui.qApp.tr("Cannot open database"),
            QtGui.qApp.tr("Unable to establish a database connection.\n"
            "This example needs SQLite support. Please read "
            "the Qt SQL driver documentation for information "
            "how to build it.\n\n" "Click Cancel to exit."),
            QtGui.QMessageBox.Cancel)
    else:
        login, password = showLoginDialog()
        
        (stoken,user_id) = vk_auth.auth(login, password, '5466274', 'audio')
        
        #QtGui.QMessageBox.about(QtGui.QWidget(), "Супер токен", stoken)
        
        #auth_session = vk.AuthSession(app_id='5466274', user_login=login, user_password=password)
        #stoken, _ = auth_session.get_access_token()
        
        print(stoken)
        
        Dialog = QtGui.QDialog()
        ui = main_form.Ui_Dialog()
        ui.setupUi(Dialog)
        
        createProgress(Dialog)
        Dialog.show()
        
        session = vk.Session(access_token=stoken)
        vkapi = vk.API(session, lang='ru')
        
        #getUsersInThread(vkapi, 157, onLoadComplete)
        
        getMusicForUsers(vkapi, ['18727902', '87550013', '273280119'], None)
        
        #getUsersFromCity(vkapi, 157, onLoadComplete)
        #count = songdb.getUsersCount()
        
        #ui.labelPeople.setText("Люди (" + str(count) + ")")

        #progress.hide()
    
        #ui.peopleFilterButton.clicked.connect(refreshClicked)
    
        #users = songdb.selectUsers()
        #ui.listPeople.setModel(users)
        
        #ui.listPeople.clicked.connect(peopleListClicked)
	
    sys.exit(app.exec_())