1313## limitations under the License.
1414
1515
16+ from collections import Counter
1617from collections import defaultdict
17- from enum import Enum
1818
19- from qtpy import QtCore , QtGui
19+ from Qt import QtCore , QtGui
2020
2121from PyFlow .Core .Common import *
2222
@@ -27,26 +27,16 @@ class InputActionType(Enum):
2727
2828
2929class InputAction (object ):
30- def __init__ (
31- self ,
32- name = "defaultName" ,
33- actionType = InputActionType .Keyboard ,
34- group = "default" ,
35- mouse = QtCore .Qt .NoButton ,
36- key = None ,
37- modifiers = QtCore .Qt .NoModifier ,
38- ):
30+ def __init__ (self , name = "defaultName" , actionType = InputActionType .Keyboard , group = "default" , mouse = QtCore .Qt .NoButton , key = None , modifiers = QtCore .Qt .NoModifier ):
3931 self .__actionType = actionType
4032 self ._name = name
4133 self ._group = group
4234 self .__data = {"mouse" : mouse , "key" : key , "modifiers" : modifiers }
4335
4436 def __str__ (self ):
45- return "{0} {1} {2}" .format (
46- QtGui .QKeySequence (self .getModifiers ()).toString (),
47- self .getMouseButton ().name .decode ("utf=8" ),
48- QtGui .QKeySequence (self .getKey ()).toString (),
49- )
37+ return "{0} {1} {2}" .format (QtGui .QKeySequence (self .getModifiers ()).toString (),
38+ self .getMouseButton ().name .decode ('utf=8' ),
39+ QtGui .QKeySequence (self .getKey ()).toString ())
5040
5141 @property
5242 def group (self ):
@@ -63,7 +53,10 @@ def __eq__(self, other):
6353 om = other .getData ()["mouse" ]
6454 ok = other .getData ()["key" ]
6555 omod = other .getData ()["modifiers" ]
66- return all ([sm == om , sk == ok , smod == omod ])
56+ smod == omod
57+ return all ([sm == om ,
58+ sk == ok ,
59+ smod == omod ])
6760
6861 def __ne__ (self , other ):
6962 sm = self .__data ["mouse" ]
@@ -72,7 +65,9 @@ def __ne__(self, other):
7265 om = other .getData ()["mouse" ]
7366 ok = other .getData ()["key" ]
7467 omod = other .getData ()["modifiers" ]
75- return not all ([sm == om , sk == ok , smod == omod ])
68+ return not all ([sm == om ,
69+ sk == ok ,
70+ smod == omod ])
7671
7772 def getName (self ):
7873 return self ._name
@@ -81,16 +76,14 @@ def getData(self):
8176 return self .__data
8277
8378 def setMouseButton (self , btn ):
84- assert isinstance (btn , QtCore .Qt .MouseButton )
79+ assert ( isinstance (btn , QtCore .Qt .MouseButton ) )
8580 self .__data ["mouse" ] = btn
8681
8782 def getMouseButton (self ):
8883 return self .__data ["mouse" ]
8984
90- def setKey (self , key = None ):
91- if key is None :
92- key = []
93- assert isinstance (key , QtCore .Qt .Key )
85+ def setKey (self , key = []):
86+ assert (isinstance (key , QtCore .Qt .Key ))
9487 self .__data ["key" ] = key
9588
9689 def getKey (self ):
@@ -119,24 +112,24 @@ def _modifiersToList(mods):
119112 result .append (QtCore .Qt .GroupSwitchModifier )
120113 return result
121114
122- @staticmethod
123- def _listOfModifiersToEnum (modifiersList ):
115+ def _listOfModifiersToEnum (self , modifiersList ):
124116 result = QtCore .Qt .NoModifier
125117 for mod in modifiersList :
126118 result = result | mod
127119 return result
128120
129121 def toJson (self ):
130- saveData = {"name" : self ._name ,
131- "group" : self ._group ,
132- "mouse" : int (self .__data ["mouse" ].value ),
133- "actionType" : self .actionType .value }
122+ saveData = {}
123+ saveData ["name" ] = self ._name
124+ saveData ["group" ] = self ._group
125+ saveData ["mouse" ] = int (self .__data ["mouse" ])
126+ saveData ["actionType" ] = self .actionType .value
127+
134128 key = self .__data ["key" ]
135129 saveData ["key" ] = int (key ) if key is not None else None
136130
137131 modifiersList = self ._modifiersToList (self .__data ["modifiers" ])
138-
139- saveData ["modifiers" ] = [i .value for i in modifiersList ]
132+ saveData ["modifiers" ] = [int (i ) for i in modifiersList ]
140133 return saveData
141134
142135 def fromJson (self , jsonData ):
@@ -145,12 +138,8 @@ def fromJson(self, jsonData):
145138 self ._group = jsonData ["group" ]
146139 self .__data ["mouse" ] = QtCore .Qt .MouseButton (jsonData ["mouse" ])
147140 keyJson = jsonData ["key" ]
148- self .__data ["key" ] = (
149- QtCore .Qt .Key (keyJson ) if isinstance (keyJson , int ) else None
150- )
151- self .__data ["modifiers" ] = self ._listOfModifiersToEnum (
152- [QtCore .Qt .KeyboardModifier (i ) for i in jsonData ["modifiers" ]]
153- )
141+ self .__data ["key" ] = QtCore .Qt .Key (keyJson ) if isinstance (keyJson , int ) else None
142+ self .__data ["modifiers" ] = self ._listOfModifiersToEnum ([QtCore .Qt .KeyboardModifier (i ) for i in jsonData ["modifiers" ]])
154143 self .__actionType = InputActionType (jsonData ["actionType" ])
155144 return self
156145 except :
0 commit comments