LinVAM/main.py

#!/usr/bin/env python3
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from ui_mainwnd import Ui_MainWidget
from profileeditwnd import ProfileEditWnd
import json
from profileexecutor import ProfileExecutor
import sys
import pickle
import time
import keyboard
import threading

class MainWnd(QWidget):
    def __init__(self, p_parent = None):
        super().__init__(p_parent)
        self.ui = Ui_MainWidget()
        self.ui.setupUi(self)
        self.m_profileExecutor = ProfileExecutor()

        self.ui.profileCbx.currentIndexChanged.connect(self.slotProfileChanged)
        self.ui.addBut.clicked.connect(self.slotAddNewProfile)
        self.ui.editBut.clicked.connect(self.slotEditProfile)
        self.ui.removeBut.clicked.connect(self.slotRemoveProfile)
        self.ui.listeningChk.stateChanged.connect(self.slotListeningEnabled)
        self.ui.ok.clicked.connect(self.slotOK)
        self.ui.cancel.clicked.connect(self.slotCancel)

        if self.loadFromDatabase() > 0 :
        # if self.loadTestProfiles() > 0:
            w_jsonProfile = self.ui.profileCbx.itemData(0)
            if w_jsonProfile != None:
                self.m_activeProfile = json.loads(w_jsonProfile)
                self.m_profileExecutor.setProfile(self.m_activeProfile)
                self.m_profileExecutor.start()

    def saveToDatabase(self):
        w_profiles = []
        w_profileCnt = self.ui.profileCbx.count()
        for w_idx in range(w_profileCnt):
            w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
            if w_jsonProfile == None:
                continue

            w_profile = json.loads(w_jsonProfile)
            w_profiles.append(w_profile)

        with open("profiles.dat", "wb") as f:
            pickle.dump(w_profiles, f, pickle.HIGHEST_PROTOCOL)

    def loadFromDatabase(self):
        w_profiles = []
        with open("profiles.dat", "rb") as f:
            w_profiles = pickle.load(f)

        for w_profile in w_profiles:
            self.ui.profileCbx.addItem(w_profile['name'])
            w_jsonProfile = json.dumps(w_profile)
            self.ui.profileCbx.setItemData(self.ui.profileCbx.count() - 1, w_jsonProfile)

        return len(w_profiles)

    def loadTestProfiles(self):

        w_carProfileDict = {
            "name": "car game",
            "commands": [
                {'name': 'forward',
                 'actions': [
                     {'name': 'key action', 'key': 'up', 'type': 1}
                 ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 12
                 },
                {'name': 'back',
                 'actions': [
                     {'name': 'key action', 'key': 'down', 'type': 1}
                 ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 3
                 },
                {'name': 'left',
                 'actions': [{'name': 'key action', 'key': 'right', 'type': 0},
                             {'name': 'key action', 'key': 'left', 'type': 1},
                             ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 10
                 },
                {'name': 'right',
                 'actions': [{'name': 'key action', 'key': 'left', 'type': 0},
                             {'name': 'key action', 'key': 'right', 'type': 1},
                             ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 3
                 },
                {'name': 'stop',
                 'actions': [
                     {'name': 'key action', 'key': 'left', 'type': 0},
                     {'name': 'key action', 'key': 'right', 'type': 0},
                     {'name': 'key action', 'key': 'up', 'type': 0},
                     {'name': 'key action', 'key': 'down', 'type': 0}
                 ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 8
                 }
            ]
        }

        w_airplaneProfileDict = {
            "name": "airplane game",
            "commands": [
                {'name': 'up',
                 'actions': [
                     {'name': 'command stop action', 'command name': 'down'},
                     {'name': 'mouse move action', 'x': 0, 'y': -5, 'absolute': False},
                     {'name': 'pause action', 'time': 0.01}
                 ],
                 'repeat': -1,
                 'async': True,
                 'threshold': 3
                 },
                {'name': 'left',
                 'actions': [
                     {'name': 'command stop action', 'command name': 'right'},
                     {'name': 'mouse move action', 'x': -5, 'y': 0, 'absolute': False},
                     {'name': 'pause action', 'time': 0.005}
                 ],
                 'repeat': -1,
                 'async': True,
                 'threshold': 3
                 },
                {'name': 'right',
                 'actions': [
                     {'name': 'command stop action', 'command name': 'left'},
                     {'name': 'mouse move action', 'x': 5, 'y': 0, 'absolute': False},
                     {'name': 'pause action', 'time': 0.005}
                 ],
                 'repeat': -1,
                 'async': True,
                 'threshold': 3
                 },
                {'name': 'down',
                 'actions': [
                     {'name': 'command stop action', 'command name': 'up'},
                     {'name': 'mouse move action', 'x': 0, 'y': 5, 'absolute': False},
                     {'name': 'pause action', 'time': 0.005}
                 ],
                 'repeat': -1,
                 'async': True,
                 'threshold': 3
                 },
                {'name': 'shoot',
                 'actions': [
                     {'name': 'mouse click action', 'button': 'left', 'type': 1},
                     {'name': 'pause action', 'time': 0.03}
                 ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 3
                 },
                {'name': 'stop',
                 'actions': [
                     {'name': 'command stop action', 'command name': 'up'},
                     {'name': 'command stop action', 'command name': 'left'},
                     {'name': 'command stop action', 'command name': 'right'},
                     {'name': 'command stop action', 'command name': 'down'},
                     {'name': 'mouse click action', 'button': 'left', 'type': 0}
                 ],
                 'repeat': 1,
                 'async': False,
                 'threshold': 3
                 }
            ]
        }
        w_profiles = []
        w_profiles.append(w_carProfileDict)
        w_profiles.append(w_airplaneProfileDict)

        i = 0
        for w_profile in w_profiles:
            self.ui.profileCbx.addItem(w_profile['name'])
            w_jsonProfile = json.dumps(w_profile)
            self.ui.profileCbx.setItemData(i, w_jsonProfile)
            i = i + 1

        return i

    def slotProfileChanged(self, p_idx):
        w_jsonProfile = self.ui.profileCbx.itemData(p_idx)
        if w_jsonProfile != None:
            self.m_activeProfile = json.loads(w_jsonProfile)
            self.m_profileExecutor.setProfile(self.m_activeProfile)

    def slotAddNewProfile(self):
        w_profileEditWnd = ProfileEditWnd(None, self)
        if w_profileEditWnd.exec() == QDialog.Accepted:
            w_profile = w_profileEditWnd.m_profile
            self.m_profileExecutor.setProfile(w_profile)
            self.ui.profileCbx.addItem(w_profile['name'])
            w_jsonProfile = json.dumps(w_profile)
            self.ui.profileCbx.setItemData(self.ui.profileCbx.count()-1, w_jsonProfile)

    def slotEditProfile(self):
        w_idx = self.ui.profileCbx.currentIndex()
        w_jsonProfile = self.ui.profileCbx.itemData(w_idx)
        w_profile = json.loads(w_jsonProfile)
        w_profileEditWnd = ProfileEditWnd(w_profile, self)
        if w_profileEditWnd.exec() == QDialog.Accepted:
            w_profile = w_profileEditWnd.m_profile
            self.m_profileExecutor.setProfile(w_profile)
            self.ui.profileCbx.setItemText(w_idx, w_profile['name'])
            w_jsonProfile = json.dumps(w_profile)
            self.ui.profileCbx.setItemData(w_idx, w_jsonProfile)

    def slotRemoveProfile(self):
        w_curIdx = self.ui.profileCbx.currentIndex()
        if w_curIdx >= 0:
            self.ui.profileCbx.removeItem(w_curIdx)

        w_curIdx = self.ui.profileCbx.currentIndex()
        if w_curIdx >= 0:
            w_jsonProfile = self.ui.profileCbx.itemData(w_curIdx)
            w_profile = json.loads(w_jsonProfile)
            self.m_profileExecutor.setProfile(w_profile)

    def slotListeningEnabled(self, p_enabled):
        if p_enabled:
            self.m_profileExecutor.setEnableListening(True)
        else:
            self.m_profileExecutor.setEnableListening(False)

    def slotOK(self):
        self.saveToDatabase()
        self.m_profileExecutor.stop()
        self.close()

    def slotCancel(self):
        self.m_profileExecutor.stop()
        self.close()
        exit()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    mainWnd = MainWnd()
    mainWnd.show()
    sys.exit(app.exec_())