def train_model(self, UI=None):
if self.k <= 0 and UI is not None:
UI.msg = QMessageBox()
UI.msg.setIcon(QMessageBox.Warning)
UI.msg.setWindowTitle("Warning")
UI.msg.setText(
"Anda harus membagi data menggunakan k-fold terlebih dahulu")
UI.msg.setStandardButtons(QMessageBox.Ok)
UI.msg.show()
UI.statusBar().showMessage("Train and test failed")
return False
self.attrs, self.clfs, self.tfidfs = ([0 for i in range(self.k)
], [0 for i in range(self.k)],
[0 for i in range(self.k)])
el = QEventLoop()
for i in range(self.k):
if UI is not None:
UI.logOutput.append(f"Train tree {i + 1}")
params = {'i': i, 'remove_zero_tfidf': True, 'UI': UI}
worker = Worker(self.mltrain_fn, params)
worker.signals.result.connect(self.mltrain_result)
self.threadpool.start(worker)
self.threadpool.waitForDone()
el.processEvents()
el.exit()
if UI is not None:
UI.logOutput.append("Training completed")
return self.attrs
class MatrixDialog(WindowModalDialog):
def __init__(self, parent):
super(MatrixDialog, self).__init__(parent)
self.setWindowTitle(_("Trezor Matrix Recovery"))
self.num = 9
self.loop = QEventLoop()
vbox = QVBoxLayout(self)
vbox.addWidget(WWLabel(MATRIX_RECOVERY))
grid = QGridLayout()
grid.setSpacing(0)
self.char_buttons = []
for y in range(3):
for x in range(3):
button = QPushButton('?')
button.clicked.connect(partial(self.process_key, ord('1') + y * 3 + x))
grid.addWidget(button, 3 - y, x)
self.char_buttons.append(button)
vbox.addLayout(grid)
self.backspace_button = QPushButton("<=")
self.backspace_button.clicked.connect(partial(self.process_key, Qt.Key_Backspace))
self.cancel_button = QPushButton(_("Cancel"))
self.cancel_button.clicked.connect(partial(self.process_key, Qt.Key_Escape))
buttons = Buttons(self.backspace_button, self.cancel_button)
vbox.addSpacing(40)
vbox.addLayout(buttons)
self.refresh()
self.show()
def refresh(self):
for y in range(3):
self.char_buttons[3 * y + 1].setEnabled(self.num == 9)
def is_valid(self, key):
return key >= ord('1') and key <= ord('9')
def process_key(self, key):
self.data = None
if key == Qt.Key_Backspace:
self.data = '\010'
elif key == Qt.Key_Escape:
self.data = 'x'
elif self.is_valid(key):
self.char_buttons[key - ord('1')].setFocus()
self.data = '%c' % key
if self.data:
self.loop.exit(0)
def keyPressEvent(self, event):
self.process_key(event.key())
if not self.data:
QDialog.keyPressEvent(self, event)
def get_matrix(self, num):
self.num = num
self.refresh()
self.loop.exec_()
class MatrixDialog(WindowModalDialog):
def __init__(self, parent):
super(MatrixDialog, self).__init__(parent)
self.setWindowTitle(_("Trezor Matrix Recovery"))
self.num = 9
self.loop = QEventLoop()
vbox = QVBoxLayout(self)
vbox.addWidget(WWLabel(MATRIX_RECOVERY))
grid = QGridLayout()
grid.setSpacing(0)
self.char_buttons = []
for y in range(3):
for x in range(3):
button = QPushButton('?')
button.clicked.connect(partial(self.process_key, ord('1') + y * 3 + x))
grid.addWidget(button, 3 - y, x)
self.char_buttons.append(button)
vbox.addLayout(grid)
self.backspace_button = QPushButton("<=")
self.backspace_button.clicked.connect(partial(self.process_key, Qt.Key_Backspace))
self.cancel_button = QPushButton(_("Cancel"))
self.cancel_button.clicked.connect(partial(self.process_key, Qt.Key_Escape))
buttons = Buttons(self.backspace_button, self.cancel_button)
vbox.addSpacing(40)
vbox.addLayout(buttons)
self.refresh()
self.show()
def refresh(self):
for y in range(3):
self.char_buttons[3 * y + 1].setEnabled(self.num == 9)
def is_valid(self, key):
return key >= ord('1') and key <= ord('9')
def process_key(self, key):
self.data = None
if key == Qt.Key_Backspace:
self.data = '\010'
elif key == Qt.Key_Escape:
self.data = 'x'
elif self.is_valid(key):
self.char_buttons[key - ord('1')].setFocus()
self.data = '%c' % key
if self.data:
self.loop.exit(0)
def keyPressEvent(self, event):
self.process_key(event.key())
if not self.data:
QDialog.keyPressEvent(self, event)
def get_matrix(self, num):
self.num = num
self.refresh()
self.loop.exec_()
class ang(QAxWidget):
def __init__(self):
# print("kiwoom() class start")
print("hello")
super().__init__()
self.login_event_loop = None
self.calculator_event_loop = QEventLoop()
self.all_stock_dict = {}
self.account_stock_dict = {}
self.get_ocx_instance()
self.event_slots()
self.signal_login_commConnect()
min_data = []
def get_ocx_instance(self):
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
def event_slots(self):
self.OnEventConnect.connect(self.login_slot)
self.OnReceiveMsg.connect(self.msg_slot)
def signal_login_commConnect(self):
self.dynamicCall("CommConnect()")
self.login_event_loop = QEventLoop()
self.login_event_loop.exec_()
def login_slot(self, err_code):
self.login_event_loop.exit()
def stop_screen_cancel(self, sScrNo=None):
self.dynamicCall("DisconnectRealData(QString)", sScrNo)
def msg_slot(self, sScrNo, sRQName, sTrCode, msg):
pass
def get_min_data(self, code, tick, sPrevNext):
self.dynamicCall("SetInputValue(QString, QString)", "종목코드", code)
self.dynamicCall("SetInputValue(QString, QString)", "틱범위", tick)
self.dynamicCall("SetInputValue(QString, QString)", "수정주가구분", "1")
self.dynamicCall("CommRqData(QString,QString,int,QString)", "주식분봉차트조회",
"opt10080", sPrevNext, "2000")
self.calculator_event_loop.exec_()
return self.min_data
def tr_data_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
if sRQName == "주식분봉차트조회":
code = self.dynamicCall(
"GetCommData(QString, QString, int , QString)", sTrCode,
sRQName, 0, "종목코드")
data = self.dynamicCall("GetCommDataEx(QString, QString)", sTrCode,
sRQName)
self.min_data = data
self.calculator_event_loop.exit()
class TheReal(QAxWidget):
def __init__(self):
super().__init__()
print(">> class TheReal() Start.")
self.realType = RealType()
self.real_screen_number = "1000"
self.login_event_loop = QEventLoop()
self.get_ocx_instance()
self.event_slots()
self.real_event_slot()
self.login_signal()
self.real_signal()
def get_ocx_instance(self):
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
def event_slots(self):
self.OnEventConnect.connect(self.login_slot)
def real_event_slot(self):
self.OnReceiveRealData.connect(self.realdata_slot)
def login_signal(self):
self.dynamicCall("CommConnect()")
self.login_event_loop.exec_()
def login_slot(self, error_code):
print(errors(error_code)[1])
login_status = self.dynamicCall("GetConnectState()")
print(">> 로그인 상태(0:연결안됨, 1:연결): %s" % login_status)
self.login_event_loop.exit()
def real_signal(self):
print(">>>>> real_signal %s" %
self.realType.REAL_TYPE["장시작시간"]["장운영구분"])
value = self.dynamicCall("SetRealReg(QString,QString,QString,QString)",
self.real_screen_number, ' ',
self.realType.REAL_TYPE["장시작시간"]["장운영구분"],
"0")
print(value)
def realdata_slot(self, stock_code, real_type, real_data):
print(">> realdata_slot start.")
print(">> stock_code: [%s]" % stock_code)
print(">> real_type: [%s]" % real_type)
print(">> real_data: [%s]" % real_data)
value = self.dynamicCall("GetCommRealData(QString,int)", stock_code,
self.realType.REAL_TYPE[real_type]["장운영구분"])
print(">> value: %" % value)
def procedure(self, spc_path=None, opened_window=None):
if opened_window is None:
return None, "Error in 'set cell free position' action. No window is opened."
if opened_window.window_type == "ms":
# cfp 設定（イベントループ）
continue_process = ["continue"] # closeEvent関数内で、参照渡しできるように、リストにしておく。
loop = QEventLoop()
def closeEvent1(self, event=None):
bool_popup = popups.WarningPopup("Closing the window will stop the batch.\nDo you really want to close?", title="warning", p_type="Bool")
done = bool_popup.exec_()
if done == 0x00004000: # Yes
continue_process[0] = "Cell free position was not set."
self.toolbar_layout.cfp_setting_popup.close()
event.accept()
loop.exit()
elif done == 0x00010000: # No
event.ignore()
def closeEvent2(self, event=None):
close_or_not = opened_window.close()
if close_or_not:
event.accept()
else:
event.ignore()
def cancelEvent():
close_or_not = opened_window.close()
if close_or_not:
loop.exit()
else:
opened_window.toolbar_layout.set_CellFreePosition()
opened_window.closeEvent = closeEvent1.__get__(opened_window, type(opened_window))
result = opened_window.toolbar_layout.set_CellFreePosition()
# cell free position が指定された時
if result:
pass
# cell free position の window で cancel された時
else:
opened_window.closeEvent = closeEvent.__get__(opened_window, type(opened_window))
loop.exit(0)
return None, "Cell free position was not set."
opened_window.toolbar_layout.cfp_setting_popup.closeEvent = closeEvent2.__get__(opened_window.toolbar_layout.cfp_setting_popup, type(opened_window.toolbar_layout.cfp_setting_popup))
opened_window.toolbar_layout.cfp_setting_popup.btnOK.clicked.connect(lambda future: loop.exit())
opened_window.toolbar_layout.cfp_setting_popup.btnCancel.clicked.connect(cancelEvent)
loop.exec_()
opened_window.closeEvent = closeEvent.__get__(opened_window, type(opened_window))
return opened_window, continue_process[0]
else:
# opened_window.close()
return None, "Invalid window type."
class Test(QAxWidget):
ge = None
def __init__(self):
super().__init__()
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
self.OnEventConnect.connect(self.login_slot)
self.OnReceiveConditionVer.connect(self.ce_slot)
self.dynamicCall("CommConnect()")
self.ge = QEventLoop()
self.ge.exec_()
self.dynamicCall("GetConditionLoad()")
def ce_slot(self, f, a):
print(self.dynamicCall("GetConditionNameList()"))
def login_slot(self, a):
self.ge.exit()
class Main():
__kiwoom = None
__event_loop = None
__condition_list = []
__condition_stock_list = []
def __init__(self):
self.__kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1")
self.__event_loop = QEventLoop()
self.__kiwoom.OnEventConnect.connect(self.__login_slot)
self.__kiwoom.OnReceiveConditionVer.connect(self.__condition_ver_slot)
self.__kiwoom.OnReceiveTrCondition.connect(self.__send_condition_slot)
def login(self):
self.__kiwoom.dynamicCall("CommConnect()")
self.__event_loop.exec_()
def get_condition_list(self):
self.__kiwoom.dynamicCall("GetConditionLoad()")
self.__event_loop.exec_()
return self.__condition_list
def print_condition_list(self, cond):
self.__kiwoom.dynamicCall("SendCondition(QString, QString, int, int)", "2000", cond[1], cond[0], 0)
self.__event_loop.exec_()
print("{} 조건식의 검색 결과".format(cond[1]))
i = 1
for stock in self.__condition_stock_list:
print("{} ".format(stock), end="")
if i % 10 == 0:
print()
i += 1
def __condition_ver_slot(self, lRet, sMsg):
condition_name_list: str
condition_name_list = self.__kiwoom.dynamicCall("GetConditionNameList()")
'''
condition_name_list
=
인덱스^조건식이름;인덱스^조건식이름;
'''
for cond_index_name in condition_name_list[:-1].split(";"):
self.__condition_list.append(cond_index_name.split("^"))
self.__event_loop.exit()
def __login_slot(self, errNo):
self.__event_loop.exit()
def __send_condition_slot(self, sScrNo, strCodeList, strConditionName, nIndex, nNext):
self.__condition_stock_list = strCodeList.split(";")
self.__event_loop.exit()
def main():
#initializing the GUI
app = QApplication(sys.argv)
# Create and display the splash screen
splash = splashscreen.ArcaneSplashScreen('image/diamondgif.gif')
splash.show()
splash.showMessage("Arcane")
#app.processEvents()
# this event loop is needed for dispatching of Qt events
initLoop = QEventLoop()
pool = Pool(processes=1)
pool.apply_async(longInitialization, [1],
callback=lambda exitCode: initLoop.exit(exitCode))
initLoop.exec_()
# create logger
logger = logging.getLogger('mainApp')
logger.setLevel(logging.DEBUG)
# create file handler
fhandler = logging.FileHandler(LOG_FILENAME)
fhandler.setLevel(logging.DEBUG)
# create formatter
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# add formatter to fhandler
fhandler.setFormatter(formatter)
# add fhandler to logger
logger.addHandler(fhandler)
###############################
logger.info('Program started.')
mainObj = MainApplication()
mainObj.show()
splash.finish(
mainObj
) #Makes the splash screen wait until the widget mainObj is displayed before calling close() on itself.
sys.exit(app.exec_())
pass
class SphinxConverter(QObject):
"""This class converts Sphinx input to HTML. It is run in a separate
thread.
"""
# This signal clears the context of the log window.
logWindowClear = pyqtSignal()
# This signal emits messages for the log window.
logWindowText = pyqtSignal(
# A string to append to the log window.
str)
def __init__(self, parent):
super().__init__(parent)
# Use an additional thread to process Sphinx output.
self._ac = AsyncController('QThread', self)
self._ac.defaultPriority = QThread.LowPriority
self._SphinxInvocationCount = 1
def terminate(self):
# Free resources.
self._ac.terminate()
def convert(self, filePath):
# Run the builder.
errString = self._runHtmlBuilder()
# Look for the HTML output.
#
# Get an absolute path to the output path, which could be relative.
outputPath = core.config()['Sphinx']['OutputPath']
projectPath = core.config()['Sphinx']['ProjectPath']
if not os.path.isabs(outputPath):
outputPath = os.path.join(projectPath, outputPath)
# Create an htmlPath as OutputPath + remainder of filePath.
htmlPath = os.path.join(outputPath + filePath[len(projectPath):])
html_file_suffix = '.html'
try:
with codecs.open(os.path.join(projectPath,
'sphinx-enki-info.txt')) as f:
hfs = f.read()
# If the file is empty, then html_file_suffix wasn't defined
# or is None. In this case, use the default extension.
# Otherwise, use the extension read from the file.
if hfs:
html_file_suffix = hfs
except:
errString = "Warning: assuming .html extension. Use " + \
"the conf.py template to set the extension.\n" + errString
pass
# First place to look: file.html. For example, look for foo.py
# in foo.py.html.
htmlFile = htmlPath + html_file_suffix
# Second place to look: file without extension.html. For
# example, look for foo.html for foo.rst.
htmlFileAlter = os.path.splitext(htmlPath)[0] + html_file_suffix
# Check that the output file produced by Sphinx is newer than
# the source file it was built from.
if os.path.exists(htmlFile):
return _checkModificationTime(filePath, htmlFile, errString)
elif os.path.exists(htmlFileAlter):
return _checkModificationTime(filePath, htmlFileAlter, errString)
else:
return (filePath,
'No preview for this type of file.<br>Expected ' +
htmlFile + " or " + htmlFileAlter, errString, QUrl())
def _runHtmlBuilder(self):
# Build the commond line for Sphinx.
if core.config()['Sphinx']['AdvancedMode']:
htmlBuilderCommandLine = core.config()['Sphinx']['Cmdline']
if sys.platform.startswith('linux'):
# If Linux is used, then subprocess cannot take the whole
# commandline as the name of an executable file. Module shlex
# has to be used to parse commandline.
htmlBuilderCommandLine = shlex.split(htmlBuilderCommandLine)
else:
# For available builder options, refer to: http://sphinx-doc.org/builders.html
htmlBuilderCommandLine = [
core.config()['Sphinx']['Executable'],
# Place doctrees in the ``_build`` directory; by default, Sphinx
# places this in _build/html/.doctrees.
'-d',
os.path.join('_build', 'doctrees'),
# Source directory -- the current directory, since we'll chdir to
# the project directory before executing this.
'.',
# Build directory
core.config()['Sphinx']['OutputPath']
]
# Invoke it.
try:
# Clear the log at the beginning of a Sphinx build.
self.logWindowClear.emit()
cwd = core.config()['Sphinx']['ProjectPath']
# If the command line is already a string (advanced mode), just print it.
# Otherwise, it's a list that should be transformed to a string.
if isinstance(htmlBuilderCommandLine, str):
htmlBuilderCommandLineStr = htmlBuilderCommandLine
else:
htmlBuilderCommandLineStr = ' '.join(htmlBuilderCommandLine)
self.logWindowText.emit('{} : {}\n\n'.format(
cwd, htmlBuilderCommandLineStr))
# Run Sphinx, reading stdout in a separate thread.
self._qe = QEventLoop()
# Sphinx will output just a carriage return (0x0D) to simulate a
# single line being updated by build status and the build
# progresses. Without universal newline support here, we'll wait
# until the build is complete (with a \n\r) to report any build
# progress! So, enable universal newlines, so that each \r will be
# treated as a separate line, providing immediate feedback on build
# progress.
popen = open_console_output(htmlBuilderCommandLine,
cwd=cwd,
universal_newlines=True)
# Perform reads in an event loop. The loop is exit when all reads
# have completed. We can't simply start the _stderr_read thread
# here, because calls to self._qe_exit() will be ignored until
# we're inside the event loop.
QTimer.singleShot(0, lambda: self._popen_read(popen))
self._qe.exec_()
except OSError as ex:
return ('Failed to execute HTML builder:\n'
'{}\n'.format(str(ex)) +
'Go to Settings -> Settings -> CodeChat to set HTML'
' builder configurations.')
return self._stderr
# Read from stdout (in this thread) and stderr (in another thread),
# so that the user sees output as the build progresses, rather than only
# producing output after the build is complete.
def _popen_read(self, popen):
# Read are blocking; we can't read from both stdout and stderr in the
# same thread without possible buffer overflows. So, use this thread to
# read from and immediately report progress from stdout. In another
# thread, read all stderr and report that after the build finishes.
self._ac.start(None, self._stderr_read, popen.stderr)
# Read a line of stdout then report it to the user immediately.
s = popen.stdout.readline()
while s:
self.logWindowText.emit(s.rstrip('\n'))
s = popen.stdout.readline()
self._SphinxInvocationCount += 1
# I would expect the following code to do the same thing. It doesn't:
# instead, it waits until Sphinx completes before returning anything.
# ???
#
# .. code-block: python
# :linenos:
#
# for s in popen.stdout:
# self.logWindowText.emit(s)
# Runs in a separate thread to read stdout. It then exits the QEventLoop as
# a way to signal that stderr reads have completed.
def _stderr_read(self, stderr):
self._stderr = stderr.read()
self._qe.exit()
class InstallWizard(QDialog, MessageBoxMixin, BaseWizard):
accept_signal = pyqtSignal()
def __init__(self, config: 'SimpleConfig', app: QApplication,
plugins: 'Plugins', *, gui_object: 'ElectrumGui'):
QDialog.__init__(self, None)
BaseWizard.__init__(self, config, plugins)
self.setWindowTitle('Electrum-LTC - ' + _('Install Wizard'))
self.app = app
self.config = config
self.gui_thread = gui_object.gui_thread
self.setMinimumSize(600, 400)
self.accept_signal.connect(self.accept)
self.title = QLabel()
self.main_widget = QWidget()
self.back_button = QPushButton(_("Back"), self)
self.back_button.setText(
_('Back') if self.can_go_back() else _('Cancel'))
self.next_button = QPushButton(_("Next"), self)
self.next_button.setDefault(True)
self.logo = QLabel()
self.please_wait = QLabel(_("Please wait..."))
self.please_wait.setAlignment(Qt.AlignCenter)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
outer_vbox = QVBoxLayout(self)
inner_vbox = QVBoxLayout()
inner_vbox.addWidget(self.title)
inner_vbox.addWidget(self.main_widget)
inner_vbox.addStretch(1)
inner_vbox.addWidget(self.please_wait)
inner_vbox.addStretch(1)
scroll_widget = QWidget()
scroll_widget.setLayout(inner_vbox)
scroll = QScrollArea()
scroll.setWidget(scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
icon_vbox = QVBoxLayout()
icon_vbox.addWidget(self.logo)
icon_vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addLayout(icon_vbox)
hbox.addSpacing(5)
hbox.addWidget(scroll)
hbox.setStretchFactor(scroll, 1)
outer_vbox.addLayout(hbox)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self.set_icon('electrum-ltc.png')
self.show()
self.raise_()
self.refresh_gui() # Need for QT on MacOSX. Lame.
def select_storage(
self, path,
get_wallet_from_daemon) -> Tuple[str, Optional[WalletStorage]]:
vbox = QVBoxLayout()
hbox = QHBoxLayout()
hbox.addWidget(QLabel(_('Wallet') + ':'))
name_e = QLineEdit()
hbox.addWidget(name_e)
button = QPushButton(_('Choose...'))
hbox.addWidget(button)
vbox.addLayout(hbox)
msg_label = WWLabel('')
vbox.addWidget(msg_label)
hbox2 = QHBoxLayout()
pw_e = PasswordLineEdit('', self)
pw_e.setFixedWidth(17 * char_width_in_lineedit())
pw_label = QLabel(_('Password') + ':')
hbox2.addWidget(pw_label)
hbox2.addWidget(pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
vbox.addSpacing(50)
vbox_create_new = QVBoxLayout()
vbox_create_new.addWidget(QLabel(_('Alternatively') + ':'),
alignment=Qt.AlignLeft)
button_create_new = QPushButton(_('Create New Wallet'))
button_create_new.setMinimumWidth(120)
vbox_create_new.addWidget(button_create_new, alignment=Qt.AlignLeft)
widget_create_new = QWidget()
widget_create_new.setLayout(vbox_create_new)
vbox_create_new.setContentsMargins(0, 0, 0, 0)
vbox.addWidget(widget_create_new)
self.set_layout(vbox, title=_('Electrum-LTC wallet'))
temp_storage = None # type: Optional[WalletStorage]
wallet_folder = os.path.dirname(path)
def on_choose():
path, __ = QFileDialog.getOpenFileName(self,
"Select your wallet file",
wallet_folder)
if path:
name_e.setText(path)
def on_filename(filename):
# FIXME? "filename" might contain ".." (etc) and hence sketchy path traversals are possible
nonlocal temp_storage
temp_storage = None
msg = None
if filename:
path = os.path.join(wallet_folder, filename)
wallet_from_memory = get_wallet_from_daemon(path)
try:
if wallet_from_memory:
temp_storage = wallet_from_memory.storage # type: Optional[WalletStorage]
else:
temp_storage = WalletStorage(path)
except (StorageReadWriteError, WalletFileException) as e:
msg = _('Cannot read file') + f'\n{repr(e)}'
except Exception as e:
self.logger.exception('')
msg = _('Cannot read file') + f'\n{repr(e)}'
else:
msg = _('')
self.next_button.setEnabled(temp_storage is not None)
user_needs_to_enter_password = False
if temp_storage:
if not temp_storage.file_exists():
msg =_("This file does not exist.") + '\n' \
+ _("Press 'Next' to create this wallet, or choose another file.")
elif not wallet_from_memory:
if temp_storage.is_encrypted_with_user_pw():
msg = _("This file is encrypted with a password.") + '\n' \
+ _('Enter your password or choose another file.')
user_needs_to_enter_password = True
elif temp_storage.is_encrypted_with_hw_device():
msg = _("This file is encrypted using a hardware device.") + '\n' \
+ _("Press 'Next' to choose device to decrypt.")
else:
msg = _("Press 'Next' to open this wallet.")
else:
msg = _("This file is already open in memory.") + "\n" \
+ _("Press 'Next' to create/focus window.")
if msg is None:
msg = _('Cannot read file')
msg_label.setText(msg)
widget_create_new.setVisible(
bool(temp_storage and temp_storage.file_exists()))
if user_needs_to_enter_password:
pw_label.show()
pw_e.show()
pw_e.setFocus()
else:
pw_label.hide()
pw_e.hide()
button.clicked.connect(on_choose)
button_create_new.clicked.connect(
partial(name_e.setText, get_new_wallet_name(wallet_folder)))
name_e.textChanged.connect(on_filename)
name_e.setText(os.path.basename(path))
def run_user_interaction_loop():
while True:
if self.loop.exec_() != 2: # 2 = next
raise UserCancelled()
assert temp_storage
if temp_storage.file_exists(
) and not temp_storage.is_encrypted():
break
if not temp_storage.file_exists():
break
wallet_from_memory = get_wallet_from_daemon(temp_storage.path)
if wallet_from_memory:
raise WalletAlreadyOpenInMemory(wallet_from_memory)
if temp_storage.file_exists() and temp_storage.is_encrypted():
if temp_storage.is_encrypted_with_user_pw():
password = pw_e.text()
try:
temp_storage.decrypt(password)
break
except InvalidPassword as e:
self.show_message(title=_('Error'), msg=str(e))
continue
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=repr(e))
raise UserCancelled()
elif temp_storage.is_encrypted_with_hw_device():
try:
self.run('choose_hw_device',
HWD_SETUP_DECRYPT_WALLET,
storage=temp_storage)
except InvalidPassword as e:
self.show_message(
title=_('Error'),
msg=
_('Failed to decrypt using this hardware device.'
) + '\n' +
_('If you use a passphrase, make sure it is correct.'
))
self.reset_stack()
return self.select_storage(path,
get_wallet_from_daemon)
except (UserCancelled, GoBack):
raise
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=repr(e))
raise UserCancelled()
if temp_storage.is_past_initial_decryption():
break
else:
raise UserCancelled()
else:
raise Exception('Unexpected encryption version')
try:
run_user_interaction_loop()
finally:
try:
pw_e.clear()
except RuntimeError: # wrapped C/C++ object has been deleted.
pass # happens when decrypting with hw device
return temp_storage.path, (temp_storage
if temp_storage.file_exists() else None)
def run_upgrades(self, storage: WalletStorage, db: 'WalletDB') -> None:
path = storage.path
if db.requires_split():
self.hide()
msg = _(
"The wallet '{}' contains multiple accounts, which are no longer supported since Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?"
).format(path)
if not self.question(msg):
return
file_list = db.split_accounts(path)
msg = _('Your accounts have been moved to') + ':\n' + '\n'.join(
file_list) + '\n\n' + _(
'Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
# raise now, to avoid having the old storage opened
raise UserCancelled()
action = db.get_action()
if action and db.requires_upgrade():
raise WalletFileException(
'Incomplete wallet files cannot be upgraded.')
if action:
self.hide()
msg = _("The file '{}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path)
if not self.question(msg):
if self.question(
_("Do you want to delete '{}'?").format(path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
self.data = json.loads(storage.read())
self.run(action)
for k, v in self.data.items():
db.put(k, v)
db.write(storage)
return
if db.requires_upgrade():
self.upgrade_db(storage, db)
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
self.logger.error("on_error", exc_info=exc_info)
self.show_error(str(exc_info[1]))
def set_icon(self, filename):
prior_filename, self.icon_filename = self.icon_filename, filename
self.logo.setPixmap(
QPixmap(icon_path(filename)).scaledToWidth(
60, mode=Qt.SmoothTransformation))
return prior_filename
def set_layout(self, layout, title=None, next_enabled=True):
self.title.setText("<b>%s</b>" % title if title else "")
self.title.setVisible(bool(title))
# Get rid of any prior layout by assigning it to a temporary widget
prior_layout = self.main_widget.layout()
if prior_layout:
QWidget().setLayout(prior_layout)
self.main_widget.setLayout(layout)
self.back_button.setEnabled(True)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self,
layout,
title=None,
raise_on_cancel=True,
next_enabled=True,
focused_widget=None):
self.set_layout(layout, title, next_enabled)
if focused_widget:
focused_widget.setFocus()
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled()
if result == 1:
raise GoBack from None
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
self.app.processEvents()
self.app.processEvents()
def remove_from_recently_open(self, filename):
self.config.remove_from_recently_open(filename)
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self,
header_layout=message,
is_valid=is_valid,
allow_multi=allow_multi,
config=self.config)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options):
slayout = SeedLayout(
title=message,
is_seed=is_seed,
options=options,
parent=self,
config=self.config,
)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
@wizard_dialog
def add_xpub_dialog(self,
title,
message,
is_valid,
run_next,
allow_multi=False,
show_wif_help=False):
header_layout = QHBoxLayout()
label = WWLabel(message)
label.setMinimumWidth(400)
header_layout.addWidget(label)
if show_wif_help:
header_layout.addWidget(InfoButton(WIF_HELP_TEXT),
alignment=Qt.AlignRight)
return self.text_input(title, header_layout, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d" % index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.'
)
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _(
'Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options)
@wizard_dialog
def confirm_seed_dialog(self, run_next, seed, test):
self.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.'
)
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text):
title = _("Your wallet generation seed is:")
slayout = SeedLayout(
seed=seed_text,
title=title,
msg=True,
options=['ext'],
config=self.config,
)
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind, force_disable_encrypt_cb):
pw_layout = PasswordLayout(
msg=msg,
kind=kind,
OK_button=self.next_button,
force_disable_encrypt_cb=force_disable_encrypt_cb)
pw_layout.encrypt_cb.setChecked(True)
try:
self.exec_layout(pw_layout.layout(),
focused_widget=pw_layout.new_pw)
return pw_layout.new_password(), pw_layout.encrypt_cb.isChecked()
finally:
pw_layout.clear_password_fields()
@wizard_dialog
def request_password(self, run_next, force_disable_encrypt_cb=False):
"""Request the user enter a new password and confirm it. Return
the password or None for no password."""
return self.pw_layout(MSG_ENTER_PASSWORD, PW_NEW,
force_disable_encrypt_cb)
@wizard_dialog
def request_storage_encryption(self, run_next):
playout = PasswordLayoutForHW(MSG_HW_STORAGE_ENCRYPTION)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.encrypt_cb.isChecked()
@wizard_dialog
def confirm_dialog(self, title, message, run_next):
self.confirm(message, title)
def confirm(self, message, title):
label = WWLabel(message)
vbox = QVBoxLayout()
vbox.addWidget(label)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self, **kwargs):
self.accept_signal.emit()
def waiting_dialog(self, task, msg, on_finished=None):
label = WWLabel(msg)
vbox = QVBoxLayout()
vbox.addSpacing(100)
label.setMinimumWidth(300)
label.setAlignment(Qt.AlignCenter)
vbox.addWidget(label)
self.set_layout(vbox, next_enabled=False)
self.back_button.setEnabled(False)
t = threading.Thread(target=task)
t.start()
while True:
t.join(1.0 / 60)
if t.is_alive():
self.refresh_gui()
else:
break
if on_finished:
on_finished()
def run_task_without_blocking_gui(self, task, *, msg=None):
assert self.gui_thread == threading.current_thread(
), 'must be called from GUI thread'
if msg is None:
msg = _("Please wait...")
exc = None # type: Optional[Exception]
res = None
def task_wrapper():
nonlocal exc
nonlocal res
try:
res = task()
except Exception as e:
exc = e
self.waiting_dialog(task_wrapper, msg=msg)
if exc is None:
return res
else:
raise exc
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
clayout = ChoicesLayout(message, c_titles)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
return action
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def derivation_and_script_type_gui_specific_dialog(
self,
*,
title: str,
message1: str,
choices: List[Tuple[str, str, str]],
hide_choices: bool = False,
message2: str,
test_text: Callable[[str], int],
run_next,
default_choice_idx: int = 0,
get_account_xpub=None,
) -> Tuple[str, str]:
vbox = QVBoxLayout()
if get_account_xpub:
button = QPushButton(_("Detect Existing Accounts"))
def on_account_select(account):
script_type = account["script_type"]
if script_type == "p2pkh":
script_type = "standard"
button_index = c_values.index(script_type)
button = clayout.group.buttons()[button_index]
button.setChecked(True)
line.setText(account["derivation_path"])
button.clicked.connect(lambda: Bip39RecoveryDialog(
self, get_account_xpub, on_account_select))
vbox.addWidget(button, alignment=Qt.AlignLeft)
vbox.addWidget(QLabel(_("Or")))
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
c_default_text = [x[2] for x in choices]
def on_choice_click(clayout):
idx = clayout.selected_index()
line.setText(c_default_text[idx])
clayout = ChoicesLayout(message1,
c_titles,
on_choice_click,
checked_index=default_choice_idx)
if not hide_choices:
vbox.addLayout(clayout.layout())
vbox.addWidget(WWLabel(message2))
line = QLineEdit()
def on_text_change(text):
self.next_button.setEnabled(test_text(text))
line.textEdited.connect(on_text_change)
on_choice_click(clayout) # set default text for "line"
vbox.addWidget(line)
self.exec_layout(vbox, title)
choice = c_values[clayout.selected_index()]
return str(line.text()), choice
@wizard_dialog
def line_dialog(self,
run_next,
title,
message,
default,
test,
warning='',
presets=(),
warn_issue4566=False):
vbox = QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
if warn_issue4566:
text_whitespace_normalised = ' '.join(text.split())
warn_issue4566_label.setVisible(
text != text_whitespace_normalised)
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
warn_issue4566_label = WWLabel(MSG_PASSPHRASE_WARN_ISSUE4566)
warn_issue4566_label.setVisible(False)
vbox.addWidget(warn_issue4566_label)
for preset in presets:
button = QPushButton(preset[0])
button.clicked.connect(
lambda __, text=preset[1]: line.setText(text))
button.setMinimumWidth(150)
hbox = QHBoxLayout()
hbox.addWidget(button, alignment=Qt.AlignCenter)
vbox.addLayout(hbox)
self.exec_layout(vbox, title, next_enabled=test(default))
return line.text()
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QVBoxLayout()
layout = SeedLayout(
xpub,
title=msg,
icon=False,
for_seed_words=False,
config=self.config,
)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network: 'Network'):
message = _("Electrum communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfill the same purpose only differing in "
"hardware. In most cases you simply want to let Electrum "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_('Cancel'))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
if r == 1:
nlayout = NetworkChoiceLayout(network, self.config, wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
self.config.set_key('auto_connect', network.auto_connect, True)
else:
network.auto_connect = True
self.config.set_key('auto_connect', True, True)
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QSlider(Qt.Horizontal, self)
n_edit = QSlider(Qt.Horizontal, self)
n_edit.setMinimum(2)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QLabel()
m_label = QLabel()
grid = QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require {0} signatures').format(m))
cw.set_m(m)
backup_warning_label.setVisible(cw.m != cw.n)
def on_n(n):
n_label.setText(_('From {0} cosigners').format(n))
cw.set_n(n)
m_edit.setMaximum(n)
backup_warning_label.setVisible(cw.m != cw.n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
vbox = QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(
WWLabel(
_("Choose the number of signatures needed to unlock funds in your wallet:"
)))
vbox.addLayout(grid)
vbox.addSpacing(2 * char_width_in_lineedit())
backup_warning_label = WWLabel(
_("Warning: to be able to restore a multisig wallet, "
"you should include the master public key for each cosigner "
"in all of your backups."))
vbox.addWidget(backup_warning_label)
on_n(2)
on_m(2)
self.exec_layout(vbox, _("Multi-Signature Wallet"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)
class Debugger(QObject, ComponentMixin):
name = 'Debugger'
preferences = Parameter.create(name='Preferences',
children=[{
'name': 'Reload CQ',
'type': 'bool',
'value': True
}, {
'name': 'Add script dir to path',
'type': 'bool',
'value': True
}, {
'name':
'Change working dir to script dir',
'type': 'bool',
'value': True
}])
sigRendered = pyqtSignal(dict)
sigLocals = pyqtSignal(dict)
sigTraceback = pyqtSignal(object, str)
sigFrameChanged = pyqtSignal(object)
sigLineChanged = pyqtSignal(int)
sigLocalsChanged = pyqtSignal(dict)
sigCQChanged = pyqtSignal(dict, bool)
sigDebugging = pyqtSignal(bool)
def __init__(self, parent):
super(Debugger, self).__init__(parent)
ComponentMixin.__init__(self)
self.inner_event_loop = QEventLoop(self)
self._actions = \
{'Run' : [QAction(icon('run'),
'Render',
self,
shortcut='F5',
triggered=self.render),
QAction(icon('debug'),
'Debug',
self,
checkable=True,
shortcut='ctrl+F5',
triggered=self.debug),
QAction(icon('arrow-step-over'),
'Step',
self,
shortcut='ctrl+F10',
triggered=lambda: self.debug_cmd(DbgState.STEP)),
QAction(icon('arrow-step-in'),
'Step in',
self,
shortcut='ctrl+F11',
triggered=lambda: None),
QAction(icon('arrow-continue'),
'Continue',
self,
shortcut='ctrl+F12',
triggered=lambda: self.debug_cmd(DbgState.CONT))
]}
def get_current_script(self):
return self.parent().components['editor'].get_text_with_eol()
def get_breakpoints(self):
return self.parent().components['editor'].get_breakpoints()
def compile_code(self, cq_script):
try:
module = imp.new_module('temp')
cq_code = compile(cq_script, '<string>', 'exec')
return cq_code, module
except Exception:
self.sigTraceback.emit(sys.exc_info(), cq_script)
return None, None
def _exec(self, code, locals_dict, globals_dict):
with ExitStack() as stack:
p = Path(self.parent().components['editor'].filename).abspath(
).dirname()
if self.preferences['Add script dir to path'] and p:
sys.path.insert(0, p)
stack.callback(sys.path.remove, p)
if self.preferences['Change working dir to script dir'] and p:
stack.enter_context(p)
exec(code, locals_dict, globals_dict)
def _inject_locals(self, module):
cq_objects = {}
def _show_object(obj, name=None, options={}):
if name:
cq_objects.update(
{name: SimpleNamespace(shape=obj, options=options)})
else:
cq_objects.update({
str(id(obj)):
SimpleNamespace(shape=obj, options=options)
})
def _debug(obj, name=None):
_show_object(obj, name, options=dict(color='red', alpha=0.2))
module.__dict__['show_object'] = _show_object
module.__dict__['debug'] = _debug
module.__dict__['log'] = lambda x: info(str(x))
module.__dict__['cq'] = cq
return cq_objects, set(module.__dict__) - {'cq'}
def _cleanup_locals(self, module, injected_names):
for name in injected_names:
module.__dict__.pop(name)
@pyqtSlot(bool)
def render(self):
if self.preferences['Reload CQ']:
reload_cq()
cq_script = self.get_current_script()
cq_code, module = self.compile_code(cq_script)
if cq_code is None: return
cq_objects, injected_names = self._inject_locals(module)
try:
self._exec(cq_code, module.__dict__, module.__dict__)
#remove the special methods
self._cleanup_locals(module, injected_names)
#collect all CQ objects if no explicit show_object was called
if len(cq_objects) == 0:
cq_objects = find_cq_objects(module.__dict__)
self.sigRendered.emit(cq_objects)
self.sigTraceback.emit(None, cq_script)
self.sigLocals.emit(module.__dict__)
except Exception:
self.sigTraceback.emit(sys.exc_info(), cq_script)
@pyqtSlot(bool)
def debug(self, value):
if value:
self.sigDebugging.emit(True)
self.state = DbgState.STEP
self.script = self.get_current_script()
code, module = self.compile_code(self.script)
if code is None:
self.sigDebugging.emit(False)
self._actions['Run'][1].setChecked(False)
return
cq_objects, injected_names = self._inject_locals(module)
self.breakpoints = [el[0] for el in self.get_breakpoints()]
#clear possible traceback
self.sigTraceback.emit(None, self.script)
try:
sys.settrace(self.trace_callback)
exec(code, module.__dict__, module.__dict__)
except Exception:
self.sigTraceback.emit(sys.exc_info(), self.script)
finally:
sys.settrace(None)
self.sigDebugging.emit(False)
self._actions['Run'][1].setChecked(False)
if len(cq_objects) == 0:
cq_objects = find_cq_objects(module.__dict__)
self.sigRendered.emit(cq_objects)
self._cleanup_locals(module, injected_names)
self.sigLocals.emit(module.__dict__)
else:
sys.settrace(None)
self.inner_event_loop.exit(0)
def debug_cmd(self, state=DbgState.STEP):
self.state = state
self.inner_event_loop.exit(0)
def trace_callback(self, frame, event, arg):
filename = frame.f_code.co_filename
if filename == DUMMY_FILE:
self.trace_local(frame, event, arg)
return self.trace_callback
else:
return None
def trace_local(self, frame, event, arg):
lineno = frame.f_lineno
line = self.script.splitlines()[lineno - 1]
f_id = id(frame)
if event in (DbgEevent.LINE, DbgEevent.RETURN):
if (self.state in (DbgState.STEP, DbgState.STEP_IN)) \
or (lineno in self.breakpoints):
self.sigLineChanged.emit(lineno)
self.sigFrameChanged.emit(frame)
self.sigLocalsChanged.emit(frame.f_locals)
self.sigCQChanged.emit(find_cq_objects(frame.f_locals), True)
self.inner_event_loop.exec_()
elif event in (DbgEevent.RETURN):
self.sigLocalsChanged.emit(frame.f_locals)
elif event == DbgEevent.CALL:
func_filename = frame.f_code.co_filename
if self.state == DbgState.STEP_IN and func_filename == DUMMY_FILE:
self.sigLineChanged.emit(lineno)
self.sigFrameChanged.emit(frame)
self.state = DbgState.STEP
class InstallWizard(QDialog, MessageBoxMixin, BaseWizard):
accept_signal = pyqtSignal()
synchronized_signal = pyqtSignal(str)
def __init__(self):
BaseWizard.__init__(self)
QDialog.__init__(self, None)
self.setWindowTitle('ElectrumSV')
self.language_for_seed = app_state.config.get('language')
self.setMinimumSize(600, 420)
self.accept_signal.connect(self.accept)
self.back_button = QPushButton(_(MSG_BUTTON_BACK), self)
self.back_button.setText(
_(MSG_BUTTON_BACK) if self.can_go_back() else _(MSG_BUTTON_CANCEL))
self.next_button = QPushButton(_(MSG_BUTTON_NEXT), self)
self.next_button.setDefault(True)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
self.scroll_widget = QWidget()
self.scroll_widget.setLayout(self.create_template_layout())
scroll = QScrollArea()
scroll.setWidget(self.scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
outer_vbox = QVBoxLayout(self)
outer_vbox.addWidget(scroll)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self.show()
self.raise_()
self.refresh_gui() # Need for QT on MacOSX. Lame.
def create_template_layout(self):
"""
The standard layout divides creates a three part template.
"""
self.title = QLabel()
self.main_widget = QWidget()
self.please_wait = QLabel(_("Please wait..."))
self.please_wait.setAlignment(Qt.AlignCenter)
vbox = QVBoxLayout()
vbox.addWidget(self.title)
vbox.addWidget(self.main_widget)
vbox.addStretch(1)
vbox.addWidget(self.please_wait)
vbox.addStretch(1)
self.template_hbox = QHBoxLayout()
vbox.addLayout(self.template_hbox)
return vbox
def select_storage(self, initial_path: str, is_startup=False):
if is_startup:
self._copy_electron_cash_wallets()
vbox = QVBoxLayout()
hbox = QHBoxLayout()
hbox.addWidget(QLabel(_('Wallet') + ':'))
self.name_e = QLineEdit()
hbox.addWidget(self.name_e)
button = QPushButton(_('Choose...'))
hbox.addWidget(button)
vbox.addLayout(hbox)
self.msg_label = QLabel('')
vbox.addWidget(self.msg_label)
hbox2 = QHBoxLayout()
self.pw_e = PasswordLineEdit()
self.pw_e.setMinimumWidth(200)
self.pw_label = QLabel(_('Password') + ':')
self.pw_label.setAlignment(Qt.AlignTop)
hbox2.addWidget(self.pw_label)
hbox2.addWidget(self.pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
self._set_standard_layout(vbox,
title=_('ElectrumSV wallet'),
back_text=_(MSG_BUTTON_CANCEL))
esv_wallets_dir = os.path.join(app_state.config.electrum_path(),
"wallets")
if is_startup:
def _show_copy_electron_cash_wallets_dialog(*args):
nonlocal esv_wallets_dir, ec_wallets_dir
d = WindowModalDialog(self, _("Copy Electron Cash Wallets"))
d.setMinimumWidth(400)
vbox, file_list = self._create_copy_electron_cash_wallets_layout(
ec_wallets_dir)
bbox = QDialogButtonBox(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
bbox.rejected.connect(d.reject)
bbox.accepted.connect(d.accept)
vbox.addWidget(bbox)
d.setLayout(vbox)
result = d.exec()
if result == QDialog.Accepted:
self._do_copy_electron_cash_wallets(
file_list, esv_wallets_dir, ec_wallets_dir)
_update_selected_wallet()
ec_import_text = ("<p>" + _(
"You have previously run Electron Cash and created at " +
"least one wallet with it. ElectrumSV should not be used to open these wallets "
+
"directly using the 'Choose' button, and you should instead use the 'Import' "
+ "button to help you copy them."
) + "</p>" + "<p>" + _(
"There are many reasons for this, and the simplest is that if Electron Cash "
+
"and ElectrumSV were to operate on the same wallet at the same time, then the "
+
"wallet will most likely become corrupted. It's simpler in every way to just "
+
"copy the wallet over to ElectrumSV and use a separate wallet file for each."
) + "</p>")
ec_import_icon = HelpLabel("label text", ec_import_text)
ec_import_icon.setPixmap(
QPixmap(icon_path("icons8-info.svg")).scaledToWidth(
16, Qt.SmoothTransformation))
ec_import_label = QLabel(
_("Existing Electron Cash wallets detected"))
ec_import_button = QPushButton(_("Import..."))
ec_import_button.clicked.connect(
_show_copy_electron_cash_wallets_dialog)
self.template_hbox.addWidget(ec_import_icon)
self.template_hbox.addWidget(ec_import_label)
self.template_hbox.addWidget(ec_import_button)
self.template_hbox.addStretch(1)
ec_wallets_dir = get_electron_cash_user_dir(esv_wallets_dir)
if len(self._list_user_wallets(ec_wallets_dir)) == 0:
ec_import_icon.set_help_text("<p>" + _(
"This feature is aimed at users who were " +
"already using Electron Cash and have existing wallets associated with it. "
+
"None were detected on this computer, but if you do have some stored in "
+
"places ElectrumSV does not know about you may need to copy them "
+ "yourself."
) + "</p>" + "<p>" + _(
"You should never open your existing Electron Cash wallets directly in "
+
"ElectrumSV as this can lead to them being opened in both applications at "
+ "the same time, and can result in corruption.") + "</p>")
ec_import_button.setEnabled(False)
ec_import_button.setToolTip(_("Nothing to import"))
ec_import_label.setText(_("No Electron Cash wallets detected"))
if WalletStorage.files_are_matched_by_path(initial_path):
self._storage_existing = WalletStorage(initial_path,
manual_upgrades=True)
wallet_folder = os.path.dirname(initial_path)
def _on_choose() -> None:
path, __ = QFileDialog.getOpenFileName(self,
"Select your wallet file",
wallet_folder)
if path:
self.name_e.setText(path)
def _on_filename(filename: str) -> None:
pw = False
self._storage_existing = None
# A relative path will be relative to the folder we offered in the choose dialog.
# An absolute path will not get joined to the dialog folder (no-op).
path = os.path.join(wallet_folder, filename)
if WalletStorage.files_are_matched_by_path(path):
try:
self._storage_existing = WalletStorage(
path, manual_upgrades=True)
except IOError:
self.next_button.setEnabled(False)
msg = _('Cannot read file')
else:
self.next_button.setEnabled(True)
if self._storage_existing.is_encrypted():
msg = '\n'.join([
_("This file is encrypted."),
_('Enter your password or choose another file.'),
])
pw = True
else:
msg = _("Press 'Next' to open this wallet.")
else:
msg = _("This file does not exist.")
if os.access(wallet_folder, os.W_OK):
self.next_button.setEnabled(True)
msg += "\n" + _(
"Press 'Next' to create this wallet, or choose another file."
)
self._path_new = path
else:
self.next_button.setEnabled(False)
msg += "\n" + _("You do not have write access " +
"to this folder to create a new wallet.")
self.msg_label.setText(msg)
if pw:
self.pw_label.show()
self.pw_e.show()
self.pw_e.setFocus()
else:
self.pw_label.hide()
self.pw_e.hide()
def _update_selected_wallet(
skip_pick_most_recent: bool = False) -> None:
wallet_filename = None
if is_startup and not skip_pick_most_recent and self._storage_existing is None:
esv_wallet_names = self._list_user_wallets(esv_wallets_dir)
if len(esv_wallet_names):
wallet_filename = esv_wallet_names[0]
if wallet_filename is None:
if self._storage_existing is not None:
wallet_filename = os.path.basename(
self._storage_existing.get_path())
else:
wallet_filename = os.path.basename(initial_path)
self.name_e.setText(wallet_filename)
button.clicked.connect(_on_choose)
self.name_e.textChanged.connect(_on_filename)
# We do not pick the most recent when first displaying the wizard because we want to
# treat the preselected wallet as the user's explicit choice. So a non-existent name
# should be a possible wallet creation.
_update_selected_wallet(skip_pick_most_recent=True)
while True:
if self._storage_existing is not None and not self._storage_existing.is_encrypted(
):
break
if self.loop.exec_() != 2: # 2 = next
return
if self._storage_existing is None:
break
if self._storage_existing is not None and self._storage_existing.is_encrypted(
):
password = self.pw_e.text()
try:
self._storage_existing.decrypt(password)
self.pw_e.setText('')
break
except DecryptionError:
QMessageBox.information(None, _('Error'),
_("Incorrect password"))
continue
except Exception as e:
logger.exception("decrypting storage")
QMessageBox.information(None, _('Error'), str(e))
return
return True
def _copy_electron_cash_wallets(self):
"""
Work out whether we should show UI to offer to copy the user's
Electron Cash wallets to their ElectrumSV wallet directory, and
if so, show it and give them the chance.
"""
# If the user has ElectrumSV wallets already, we do not offer to copy the one's
# Electron Cash has.
esv_wallets_dir = os.path.join(app_state.config.electrum_path(),
"wallets")
if len(self._list_user_wallets(esv_wallets_dir)) > 0:
return
ec_wallets_dir = get_electron_cash_user_dir(esv_wallets_dir)
ec_wallet_count = len(self._list_user_wallets(ec_wallets_dir))
# If the user does not have Electron Cash wallets to copy, there's no point in offering.
if ec_wallet_count == 0:
return
vbox, file_list = self._create_copy_electron_cash_wallets_layout(
ec_wallets_dir)
self._set_standard_layout(vbox,
title=_('Import Electron Cash wallets'))
v = self.loop.exec_()
# Cancel, exit application.
if v == -1:
raise UserCancelled()
if v != 2:
raise GoBack()
self._do_copy_electron_cash_wallets(file_list, esv_wallets_dir,
ec_wallets_dir)
def _do_copy_electron_cash_wallets(self, file_list, esv_wallets_dir,
ec_wallets_dir):
# If the user selected any files, then we copy them before exiting to the next page.
copy_count = 0
for item in file_list.selectedItems():
filename = item.text()
source_path = os.path.join(ec_wallets_dir, filename)
target_path = os.path.join(esv_wallets_dir, filename)
# If they are copying an Electron Cash wallet over an ElectrumSV wallet, make sure
# they confirm they are going to replace/overwrite it.
if os.path.exists(target_path):
if self.question(
_("You already have a wallet named '{}' for ElectrumSV. "
+ "Replace/overwrite it?").format(filename), self,
_("Delete Wallet?")):
os.remove(target_path)
else:
continue
try:
shutil.copyfile(source_path, target_path)
copy_count += 1
except shutil.Error:
# For now we ignore copy errors.
pass
if copy_count == 1:
self.show_message(_("1 wallet copied."))
elif copy_count > 1:
self.show_message(_("%d wallets copied.") % copy_count)
def _create_copy_electron_cash_wallets_layout(self, ec_wallets_dir):
def update_summary_label():
selection_count = len(file_list.selectedItems())
if selection_count == 0:
summary_label.setText(
_("No wallets are selected / will be copied."))
elif selection_count == 1:
summary_label.setText(
_("1 wallet is selected / will be copied."))
else:
summary_label.setText(
_("%d wallets are selected / will be copied.") %
selection_count)
wallet_filenames = sorted(os.listdir(ec_wallets_dir),
key=lambda s: s.lower())
file_list = QListWidget()
file_list.setSelectionMode(QAbstractItemView.ExtendedSelection)
for filename in wallet_filenames:
if not self._ignore_wallet_file(
os.path.join(ec_wallets_dir, filename)):
file_list.addItem(QListWidgetItem(filename))
file_list.itemSelectionChanged.connect(update_summary_label)
vbox = QVBoxLayout()
introduction_label = QLabel(
_("Your Electron Cash wallet directory was found. If you want ElectrumSV to import "
"any of them on your behalf, select the ones you want copied from the list below "
"before clicking the Next button."))
introduction_label.setWordWrap(True)
vbox.setSpacing(20)
vbox.addWidget(introduction_label)
vbox.addWidget(file_list)
summary_label = QLabel()
update_summary_label()
vbox.addWidget(summary_label)
return vbox, file_list
def _list_user_wallets(self, wallets_path):
if os.path.exists(wallets_path):
from stat import ST_MTIME
l = [(os.stat(os.path.join(wallets_path,
filename))[ST_MTIME], filename)
for filename in os.listdir(wallets_path)
if not self._ignore_wallet_file(
os.path.join(wallets_path, filename))]
l = sorted(l, reverse=True)
return [entry[1] for entry in l]
return []
def _ignore_wallet_file(self, wallet_path):
if os.path.isdir(wallet_path):
return True
if wallet_path.startswith("."):
return True
return False
def run_and_get_wallet(self) -> Optional[Abstract_Wallet]:
# NOTE(rt12): This used to have unused code related to resuming incompletely created
# wallets. This is worth supporting, but not at this stage where we will eventually
# rewrite for a new wallet wizard and multiple accounts and legacy wallets so on.
if self._storage_existing is not None:
path = self._storage_existing.get_path()
if self._storage_existing.requires_split():
msg = _(
"The wallet '{}' contains multiple accounts, which are not supported.\n\n"
"Do you want to split your wallet "
"into multiple files?").format(path)
if not MessageBox.question(msg):
return
file_list = '\n'.join(self._storage_existing.split_accounts())
msg = (_('Your accounts have been moved to') + ':\n' +
file_list + '\n\n' +
_('Do you want to delete the old file') + ':\n' + path)
if self.question(msg):
# We know that this will be the only relevant path and will not require
# extensions, as it predates the use of the database.
os.remove(path)
self.show_warning(_('The file was removed'))
return
if self._storage_existing.requires_upgrade():
msg = _(
"The format of your wallet '%s' must be upgraded for ElectrumSV. "
"This change will not be backward compatible, " +
"and your existing wallet will be backed up. Proceed?"
) % path
if not MessageBox.question(msg):
return
self._storage_existing.upgrade()
self._parent_wallet = ParentWallet(self._storage_existing)
else:
assert self._path_new is not None, "user should have selected storage already"
wallet_filename = os.path.basename(self._path_new)
# Make a temporary directory as securely as possible, where the new wallet will be
# created.
creation_path = tempfile.mkdtemp()
try:
path_new = os.path.join(creation_path, wallet_filename)
self._storage_new = WalletStorage(path_new)
# Ensure the path is full and includes the extension.
path_new = self._storage_new.get_path()
self.run("new")
if self._parent_wallet is not None:
self._parent_wallet.save_storage()
if self._parent_wallet is not None:
self._parent_wallet.move_to(self._path_new)
else:
# We want to make sure we are not keeping the database open, before deleting.
self._storage_new.close()
del self._storage_new
finally:
shutil.rmtree(creation_path)
return self._parent_wallet
def finished(self):
"""Called in hardware client wrapper, in order to close popups."""
return
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
logger.exception("")
self.show_error(str(exc_info[1]))
def _remove_layout_from_widget(self, widget):
"""
The only way to remove a layout from a first widget, is to transfer it to a second one.
This needs to be done, to be able to set a new layout on the first widget.
"""
existing_layout = widget.layout()
QWidget().setLayout(existing_layout)
def _set_layout(self, layout, next_enabled=True, back_text=None):
"""
Set a layout that is in control of the whole display area.
"""
self._remove_layout_from_widget(self.scroll_widget)
self.scroll_widget.setLayout(layout)
self.back_button.setEnabled(True)
if back_text is not None:
self.back_button.setText(back_text)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
def _set_standard_layout(self,
layout,
title=None,
next_enabled=True,
back_text=None):
"""
Ensure the standard template layout is in place.
And put the current stage's sub-layout in the defined place.
"""
self._remove_layout_from_widget(self.scroll_widget)
self.scroll_widget.setLayout(self.create_template_layout())
self.title.setText("<b>%s</b>" % title if title else "")
self.title.setVisible(bool(title))
self.main_widget.setLayout(layout)
if back_text is None:
self.back_button.setText(_(MSG_BUTTON_BACK))
else:
self.back_button.setText(back_text)
self.back_button.setEnabled(True)
self.next_button.setText(_(MSG_BUTTON_NEXT))
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self,
layout,
title=None,
raise_on_cancel=True,
next_enabled=True):
self._set_standard_layout(layout, title, next_enabled)
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled
if result == 1:
raise GoBack
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
app_state.app.processEvents()
app_state.app.processEvents()
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self,
title=message,
is_valid=is_valid,
allow_multi=allow_multi)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options):
slayout = SeedLayout(title=message,
is_seed=is_seed,
options=options,
parent=self)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
@wizard_dialog
def add_xpub_dialog(self,
title,
message,
is_valid,
run_next,
allow_multi=False):
return self.text_input(title, message, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d" % index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.'
)
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _(
'Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options)
@wizard_dialog
def confirm_seed_dialog(self, run_next, test):
app_state.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.'
)
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text):
title = _("Your wallet generation seed is:")
slayout = SeedLayout(seed=seed_text,
title=title,
msg=True,
options=['ext'])
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind):
playout = PasswordLayout(None, msg, kind, self.next_button)
self.exec_layout(playout.layout())
return playout.new_password()
@wizard_dialog
def request_password(self, run_next):
"""Request the user enter a new password and confirm it. Return
the password or None for no password."""
return self.pw_layout(MSG_ENTER_PASSWORD, PW_NEW)
@wizard_dialog
def confirm_dialog(self, title, message, run_next):
self.confirm(message, title)
def confirm(self, message, title):
label = WWLabel(message)
vbox = QVBoxLayout()
vbox.addWidget(label)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self):
self.accept_signal.emit()
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
clayout = ChoicesLayout(message, c_titles)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
return action
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def line_dialog(self, run_next, title, message, default, test, warning=''):
vbox = QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
self.exec_layout(vbox, title, next_enabled=test(default))
return ' '.join(line.text().split())
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QVBoxLayout()
layout = SeedLayout(xpub, title=msg, icon=False)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network):
message = _(
"ElectrumSV communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfil the same purpose only differing in "
"hardware. In most cases you simply want to let ElectrumSV "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_(MSG_BUTTON_CANCEL))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
app_state.config.set_key('auto_connect', r == 0, True)
if r == 1:
nlayout = NetworkChoiceLayout(network,
app_state.config,
wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QSlider(Qt.Horizontal, self)
n_edit = QSlider(Qt.Horizontal, self)
n_edit.setMinimum(2)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QLabel()
m_label = QLabel()
grid = QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require %d signatures') % m)
cw.set_m(m)
def on_n(n):
n_label.setText(_('From %d cosigners') % n)
cw.set_n(n)
m_edit.setMaximum(n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
on_n(2)
on_m(2)
vbox = QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(
WWLabel(
_("Choose the number of signatures needed to unlock "
"funds in your wallet:")))
vbox.addLayout(grid)
self.exec_layout(vbox, _("Multi-Signature Wallet"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)
class MainWindowDialog(MainWrappedUI):
# Signal used when user close app after clicked esc or cros to close app.
# If user clicked Yes on message box return True other way return False.
closingSignal = pyqtSignal(QEvent)
callSignal = pyqtSignal(bool)
creditsSignal = pyqtSignal(bool)
def __init__(self, client):
super(MainWindowDialog, self).__init__()
self.client = client
self.prefix = ":/avatars/"
self.loop = QEventLoop()
self.timer = QTimer()
self.timer.setSingleShot(True)
self.timer.timeout.connect(lambda: self.loop.exit(1))
self.table_widget_list_of_users.setIconSize(QSize(72, 72))
self.table_widget_list_of_users.setSelectionBehavior(QTableView.SelectRows)
self.set_menubar_about(self.showCredits)
self.set_push_button_logout(self.closeApp)
self.set_push_button_call(self.call_someone)
"""
self.setWindowFlags(Qt.CustomizeWindowHint)
self.setWindowFlags(
QtCore.Qt.Window |
QtCore.Qt.CustomizeWindowHint |
QtCore.Qt.WindowTitleHint |
QtCore.Qt.WindowCloseButtonHint |
QtCore.Qt.WindowStaysOnTopHint
)
self.setAttribute(Qt.WA_TranslucentBackground)
"""
# connect(self.close_event_message_box)
def showCredits(self):
self.creditsSignal.emit(True)
def closeApp(self, event):
print(event)
self.close()
def closeEvent(self, event):
self.closingSignal.emit(event)
def setUserName(self, user_name):
self.userName = user_name
def setUserAvatar(self, user_avatar):
try:
self.label_avatar.setPixmap(
QtGui.QPixmap(self.prefix + str(user_avatar))
)
except Exception as err:
print(err)
def getList(self):
payload = {"type": "d", "description": "GET"}
data = json.dumps(payload).encode("utf-8")
print("{*} MainWindow info : Sended data to server:", data)
self.client.sendMessage(data)
self.read()
def call_someone(self):
row = int(self.table_widget_list_of_users.currentRow())
if row != None:
try:
where = self.table_widget_list_of_users.item(row, 0).text()
except Exception as err:
print(err)
if where != None:
print("Wybrano dzwonienie do ", where)
payload = {"type": "d", "description": "INVITE", "call_to": where}
data = json.dumps(payload).encode("utf-8")
print(data)
self.client.sendMessage(data)
self.timer.start(5000)
self.read()
def waiting_for_signal(self):
self.timer.start(10000) # 10 second time-out
print('{*} MainWindow info: waiting for response')
if self.loop.exec_() == 0:
self.timer.stop()
print("{*} MainWindow info: stop timer")
return True
else:
print('{!} MainWindow error: time-out :(')
return False
def showConnectionStatus(self, status):
# self.addWidget(self.statusBar)
self.statusBar.showMessage(status)
def read(self):
if self.waiting_for_signal():
print("{*} MainWindow getting from Server:", self.client.received)
if self.client.status == "202 USERS":
print("{*} MainWindow users: ", self.client.users)
self.set_who_is_signed(self.client.who_signed)
print("{*} MainWindow info: Users - ",self.client.users)
self.add_row_to_list_of_users(self.client.users)
elif self.client.status == "200 INVITE":
status = "Nawiązywanie polaczenia"
self.showConnectionStatus(status)
print(status)
elif self.client.status == "406 INVITE":
status = "Nie można się połączyć z wybranym użytkownikiem"
self.showConnectionStatus(status)
print(status)
elif self.client.status == "200 END":
status = "Zakonczono polaczenie"
self.showConnectionStatus(status)
self.client.send_ack_on_close_conn()
print(status)
elif self.client.status == "402 NOT ACCEPTABLE":
status = "Aktywuj swoje konto"
self.showConnectionStatus(status)
print(status)
else:
status = "Serwer nie odpowiada"
self.showConnectionStatus(status)
print(status)
print("{!} MainWindow error: Didn't get response")
def keyPressEvent(self, event):
"""
Close application from escape key.
"""
if event.key() == Qt.Key_Escape:
self.close()
pixmap = self.movie.currentPixmap()
self.setPixmap(pixmap)
self.setMask(pixmap.mask())
# Put your initialization code here
def longInitialization(arg):
time.sleep(arg)
return 0
if __name__ == "__main__":
import sys, time
app = QApplication(sys.argv)
# Create and display the splash screen
# splash_pix = QPixmap('a.gif')
splash = MySplashScreen("images/giphy2.gif", Qt.Qt.WindowStaysOnTopHint)
# splash.setMask(splash_pix.mask())
#splash.raise_()
splash.show()
app.processEvents()
# this event loop is needed for dispatching of Qt events
initLoop = QEventLoop()
pool = Pool(processes=1)
pool.apply_async(longInitialization, [2], callback=lambda exitCode: initLoop.exit(exitCode))
initLoop.exec_()
form = Form()
form.show()
splash.finish(form)
app.exec_()
class OPT30009(QAxWidget):
def __init__(self):
super().__init__()
# 전역변수 모음
self.account_number = None
self.total_days = 0
self.stock_day_list = []
self.target_code_list = []
self.market_type = None
self.stock_day_table = "stock_day_opt30009"
# 종목 일봉 데이터 수집 화면번호 지정
self.stock_day_screen_number = "4000"
# 짤려서 현재 실행할 순번 1001, 569
# 2020-10-05 23:27
# 2020-10-06 22:16 ~
# 2020-10-08 22:58 ~ 2020-10-09 00:30 완료
# 2020-10-12 MON
# 2020-10-13 TUE 20:01 ~
self.current_day = 0
# 수집 대상 기간 일수
self.days = 0
# 로그인 Event Loop 실행
self.login_event_loop = QEventLoop()
self.day_info_event_loop = QEventLoop()
# Kiwoom OCX 사용 함수 호출
self.get_ocx_instance()
# Event Slots
self.event_slots()
# 로그인 signal 함수 호출
self.signal_login_commConnect()
# 계좌정보 가져오기
self.get_login_info()
# 전체 종목 수집 및 분석을 위한 전체 종복 정보 가져오기
self.get_day_info_by_stock(sPrevNext="0")
def get_ocx_instance(self):
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
def event_slots(self):
self.OnEventConnect.connect(self.login_event_slot)
self.OnReceiveTrData.connect(self.trdata_slot)
def signal_login_commConnect(self):
self.dynamicCall("CommConnect()")
self.login_event_loop.exec_()
def login_event_slot(self, error_code):
self.login_event_loop.exit()
def get_login_info(self):
accounts = self.dynamicCall("GetLoginInfo(QString)", "ACCNO")
self.account_number = accounts.split(";")[0]
def get_day_info_by_stock(self, sPrevNext="0"):
QTest.qWait(1000) # 3600 Delay Time
if(sPrevNext == '0'):
print("순위,종목코드,종목명,현재가,대비기호,전일대비,등락률,매도잔량,매수잔량,거래량,거래대금")
############################## opt30009 : ELW등락율순위요청 ##############################
# 정렬구분 = 1:상승률, 2:상승폭, 3:하락률, 4:하락폭
self.dynamicCall("SetInputValue(QString,QString)", "정렬구분", "1")
# 권리구분 = 000:전체, 001:콜, 002:풋, 003:DC, 004:DP, 006:조기종료콜, 007:조기종료풋
self.dynamicCall("SetInputValue(QString,QString)", "권리구분", "000")
# 거래종료제외 = 1:거래종료제외, 0:거래종료포함
self.dynamicCall("SetInputValue(QString,QString)", "거래종료제외", "1")
self.dynamicCall("CommRqData(QString,QString,int,QString)", "ELW등락율순위요청", "opt30009", sPrevNext, self.stock_day_screen_number)
self.day_info_event_loop.exec_()
def trdata_slot(self, screen_number, tr_name, tr_code, record_name, sPrevNext):
# 멀티데이터 일봉 차트 가져오기
multi_cnt = self.dynamicCall("GetRepeatCnt(QString,QString)", tr_code, tr_name)
for i in range(multi_cnt):
data = []
output01 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "순위").strip()
output02 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "종목코드").strip()
output03 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "종목명").strip()
output04 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "현재가").strip()
output05 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "대비기호").strip()
output06 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "전일대비").strip()
output07 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "등락률").strip()
output08 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "매도잔량").strip()
output09 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "매수잔량").strip()
output10 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "거래량").strip()
output11 = self.dynamicCall("GetCommData(QString,QString,int,QString)", tr_code, tr_name, i, "거래대금").strip()
self.total_days = self.total_days + 1
data.append("")
data.append(output01)
data.append(output02)
data.append(output03)
data.append(output04)
data.append(output05)
data.append(output06)
data.append(output07)
data.append(output08)
data.append(output09)
data.append(output10)
data.append(output11)
data.append("")
self.stock_day_list.append(data.copy())
if int(output11) > 0:
print("%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s" % (output01, output02, output03, output04, output05, output06, output07, output08, output09, output10, output11))
if self.total_days == self.days:
# print(">> %s,%s,%s,%s,%s,%s,%s,%s,%s " % ((i+1),stock_code,date,close_price,open_price,high_price,low_price,volume,trading_value))
sPrevNext = "0"
break
if sPrevNext == "2":
self.get_day_info_by_stock(sPrevNext=sPrevNext)
else:
self.day_info_event_loop.exit()
def insert_stock_day_list(self):
conn_string = "host='localhost' dbname='postgres' user='******' password='******' port='5432'"
conn = psycopg2.connect(conn_string)
cur = conn.cursor()
day_num = 0
for stock_day in self.stock_day_list:
delete_day_sql = "delete from "+self.stock_day_table+" where stock_code = %s and dt = %s"
cur.execute(delete_day_sql, (stock_day[1], stock_day[3],))
day_num = day_num + 1
sql = "insert into "+self.stock_day_table+" (stock_code,stock_name,day_num,dt,volume,trading_value,close_price,open_price,high_price,low_price,market_type) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
cur.execute(sql,(stock_day[1],stock_day[2],day_num,stock_day[3],stock_day[4],stock_day[5],stock_day[6],stock_day[7],stock_day[8],stock_day[9],stock_day[10]))
conn.commit()
cur.close()
conn.close()
def update_day_num(self):
conn_string = "host='localhost' dbname='postgres' user='******' password='******' port='5432'"
conn = psycopg2.connect(conn_string)
cur = conn.cursor()
# 시장유형의 모든 종목 입력 후 시장유형 별 일봉 순번 갱신
update_day_sql = " update "+self.stock_day_table+" s "
update_day_sql += " set day_num = a.day_num "
update_day_sql += " from "
update_day_sql += " ( "
update_day_sql += " select row_number() over(partition by stock_code order by dt desc) as day_num "
update_day_sql += " , stock_code "
update_day_sql += " , dt "
update_day_sql += " , market_type "
update_day_sql += " from "+self.stock_day_table+" "
update_day_sql += " where market_type = %s "
update_day_sql += " ) as a "
update_day_sql += " where s.stock_code = a.stock_code "
update_day_sql += " and s.dt = a.dt "
update_day_sql += " and s.market_type = a.market_type "
update_day_sql += " and s.market_type = %s "
cur.execute(update_day_sql, (self.market_type, self.market_type,))
conn.commit()
cur.close()
conn.close()
class InstallWizard(QDialog, MessageBoxMixin, BaseWizard,
TermsAndConditionsMixin):
accept_signal = pyqtSignal()
def __init__(self, config: 'SimpleConfig', app: QApplication,
plugins: 'Plugins', **kwargs):
QDialog.__init__(self, parent=kwargs.get('parent', None))
if kwargs.get('parent', None):
self.setWindowModality(QtCore.Qt.ApplicationModal)
BaseWizard.__init__(self, config, plugins)
self.app = app
self.config = config
if kwargs.get('minimum_size', None):
self.setMinimumSize(*kwargs.get('minimum_size'))
else:
self.setMinimumSize(600, 400)
self.accept_signal.connect(self.accept)
self.title = QLabel()
self.main_widget = QWidget()
self.back_button = QPushButton(self)
self.next_button = QPushButton(self)
self.next_button.text()
self.next_button.setDefault(True)
self.logo = QLabel()
self.please_wait = QLabel()
self.please_wait.setAlignment(Qt.AlignCenter)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
outer_vbox = QVBoxLayout(self)
inner_vbox = QVBoxLayout()
inner_vbox.addWidget(self.title)
inner_vbox.addWidget(self.main_widget)
inner_vbox.addStretch(1)
inner_vbox.addWidget(self.please_wait)
inner_vbox.addStretch(1)
scroll_widget = QWidget()
if not kwargs.get('turn_off_icon', None):
scroll_widget.setLayout(inner_vbox)
scroll = QScrollArea()
scroll.setWidget(scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
icon_vbox = QVBoxLayout()
icon_vbox.addWidget(self.logo)
icon_vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addLayout(icon_vbox)
hbox.addSpacing(5)
hbox.addWidget(scroll)
hbox.setStretchFactor(scroll, 1)
if kwargs.get('turn_off_icon', None):
outer_vbox.addLayout(inner_vbox)
else:
outer_vbox.addLayout(hbox)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self._add_advanced_button()
self.set_icon('electrum.png')
self._set_gui_text()
self.show()
self.raise_()
self.refresh_gui() # Need for QT on MacOSX. Lame.
LastChosenState.clear()
def _set_gui_text(self):
self.setWindowTitle('Electrum - ' + _('Install Wizard'))
self.back_button.setText(
_('Back') if self.can_go_back() else _('Cancel'))
self.next_button.setText(_("Next"))
self.please_wait.setText(_("Please wait..."))
self._translate_advanced_options()
def select_and_save_language(self):
"""Method for selecting and saving language in config file as {'language': <language-abbreviation: str>}"""
filtered_languages = dict(
filter(lambda item: item[1] != 'Default', languages.items()))
language_abbreviations = list(filtered_languages.keys())
default_language = get_default_language()
vbox = QVBoxLayout()
cb = QComboBox()
cb.addItems(filtered_languages.values())
cb.setCurrentIndex(language_abbreviations.index(default_language))
vbox.addWidget(cb)
def on_change():
language_abbreviation = language_abbreviations[cb.currentIndex()]
set_language(language_abbreviation)
self._set_gui_text()
self.refresh_gui()
cb.currentIndexChanged.connect(on_change)
# refresh config language
on_change()
pushed_button = self.exec_layout(
vbox, title=_('Select installation language'))
if pushed_button == PushedButton.NEXT:
self.config.set_key('language',
language_abbreviations[cb.currentIndex()])
def select_storage(
self, path,
get_wallet_from_daemon) -> Tuple[str, Optional[WalletStorage]]:
vbox = QVBoxLayout()
hbox = QHBoxLayout()
label = QLabel(
_('Enter the name of the new wallet or load an existing wallet by clicking "Choose".'
))
label.setWordWrap(True)
vbox.addWidget(label)
hbox.addWidget(QLabel(_('Wallet') + ':'))
self.name_e = QLineEdit()
hbox.addWidget(self.name_e)
button = QPushButton(_('Choose...'))
hbox.addWidget(button)
vbox.addLayout(hbox)
self.msg_label = WWLabel('')
vbox.addWidget(self.msg_label)
hbox2 = QHBoxLayout()
self.pw_e = QLineEdit('', self)
self.pw_e.setFixedWidth(17 * char_width_in_lineedit())
self.pw_e.setEchoMode(2)
self.pw_label = QLabel(_('Password') + ':')
hbox2.addWidget(self.pw_label)
hbox2.addWidget(self.pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
self.set_layout(vbox, title=_('Electrum wallet'))
temp_storage = None # type: Optional[WalletStorage]
wallet_folder = os.path.dirname(path)
def on_choose():
path, __ = QFileDialog.getOpenFileName(self,
"Select your wallet file",
wallet_folder)
if path:
self.name_e.setText(path)
def on_filename(filename):
# FIXME? "filename" might contain ".." (etc) and hence sketchy path traversals are possible
nonlocal temp_storage
temp_storage = None
msg = None
path = os.path.join(wallet_folder, filename)
wallet_from_memory = get_wallet_from_daemon(path)
try:
if wallet_from_memory:
temp_storage = wallet_from_memory.storage # type: Optional[WalletStorage]
else:
temp_storage = WalletStorage(path, manual_upgrades=True)
except (StorageReadWriteError, WalletFileException) as e:
msg = _('Cannot read file') + f'\n{repr(e)}'
except Exception as e:
self.logger.exception('')
msg = _('Cannot read file') + f'\n{repr(e)}'
self.next_button.setEnabled(temp_storage is not None)
user_needs_to_enter_password = False
if temp_storage:
if not temp_storage.file_exists():
msg = _("This file does not exist.") + '\n' \
+ _("Press 'Next' to create this wallet, or choose another file.")
elif not wallet_from_memory:
if temp_storage.is_encrypted_with_user_pw():
msg = _("This file is encrypted with a password.") + '\n' \
+ _('Enter your password or choose another file.')
user_needs_to_enter_password = True
elif temp_storage.is_encrypted_with_hw_device():
msg = _("This file is encrypted using a hardware device.") + '\n' \
+ _("Press 'Next' to choose device to decrypt.")
else:
msg = _("Press 'Next' to open this wallet.")
else:
msg = _("This file is already open in memory.") + "\n" \
+ _("Press 'Next' to create/focus window.")
if msg is None:
msg = _('Cannot read file')
self.msg_label.setText(msg)
if user_needs_to_enter_password:
self.pw_label.show()
self.pw_e.show()
self.pw_e.setFocus()
else:
self.pw_label.hide()
self.pw_e.hide()
button.clicked.connect(on_choose)
self.name_e.textChanged.connect(on_filename)
self.name_e.setText(os.path.basename(path))
while True:
if self.loop.exec_() != 2: # 2 = next
raise UserCancelled
assert temp_storage
if temp_storage.file_exists() and not temp_storage.is_encrypted():
break
if not temp_storage.file_exists():
break
wallet_from_memory = get_wallet_from_daemon(temp_storage.path)
if wallet_from_memory:
raise WalletAlreadyOpenInMemory(wallet_from_memory)
if temp_storage.file_exists() and temp_storage.is_encrypted():
if temp_storage.is_encrypted_with_user_pw():
password = self.pw_e.text()
try:
temp_storage.decrypt(password)
break
except InvalidPassword as e:
self.show_message(title=_('Error'), msg=str(e))
continue
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=repr(e))
raise UserCancelled()
elif temp_storage.is_encrypted_with_hw_device():
try:
self.run('choose_hw_device',
HWD_SETUP_DECRYPT_WALLET,
storage=temp_storage)
except InvalidPassword as e:
self.show_message(
title=_('Error'),
msg=_(
'Failed to decrypt using this hardware device.'
) + '\n' +
_('If you use a passphrase, make sure it is correct.'
))
self.reset_stack()
return self.select_storage(path,
get_wallet_from_daemon)
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=repr(e))
raise UserCancelled()
if temp_storage.is_past_initial_decryption():
break
else:
raise UserCancelled()
else:
raise Exception('Unexpected encryption version')
return temp_storage.path, (temp_storage
if temp_storage.file_exists() else None)
def run_upgrades(self, storage):
path = storage.path
if storage.requires_split():
self.hide()
msg = _(
"The wallet '{path}' contains multiple accounts, which are no longer supported since Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?"
).format(path=path)
if not self.question(msg):
return
file_list = '\n'.join(storage.split_accounts())
msg = _('Your accounts have been moved to'
) + ':\n' + file_list + '\n\n' + _(
'Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
# raise now, to avoid having the old storage opened
raise UserCancelled()
action = storage.get_action()
if action and storage.requires_upgrade():
raise WalletFileException(
'Incomplete wallet files cannot be upgraded.')
if action:
self.hide()
msg = _(
"The file '{path}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path=path)
if not self.question(msg):
if self.question(
_("Do you want to delete '{path}'?").format(
path=path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
self.data = storage.db.data # FIXME
self.run(action)
for k, v in self.data.items():
storage.put(k, v)
storage.write()
return
if storage.requires_upgrade():
self.upgrade_storage(storage)
def finished(self):
"""Called in hardware client wrapper, in order to close popups."""
return
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
self.logger.error("on_error", exc_info=exc_info)
self.show_error(str(exc_info[1]))
def set_icon(self, filename):
prior_filename, self.icon_filename = self.icon_filename, filename
self.logo.setPixmap(
QPixmap(icon_path(filename)).scaledToWidth(
60, mode=Qt.SmoothTransformation))
return prior_filename
def set_layout(self, layout, title=None, next_enabled=True):
self.title.setText("<b>%s</b>" % title if title else "")
self.title.setVisible(bool(title))
# Get rid of any prior layout by assigning it to a temporary widget
prior_layout = self.main_widget.layout()
if prior_layout:
QWidget().setLayout(prior_layout)
self.main_widget.setLayout(layout)
self.back_button.setEnabled(True)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self,
layout,
title=None,
raise_on_cancel=True,
next_enabled=True):
self.set_layout(layout, title, next_enabled)
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled
if result == 1:
raise GoBack from None
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
self.app.processEvents()
self.app.processEvents()
def remove_from_recently_open(self, filename):
self.config.remove_from_recently_open(filename)
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self,
header_layout=message,
is_valid=is_valid,
allow_multi=allow_multi)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options, on_import=False):
imported_wallet = self.wallet_type if on_import else None
slayout = SeedLayout(title=message,
is_seed=is_seed,
options=options,
parent=self,
imported_wallet=imported_wallet)
if imported_wallet == 'standard':
slayout.show_default_options()
self.exec_advanced_layout(
layout=slayout,
default_show_function=slayout.show_default_options,
advanced_show_function=slayout.show_advanced_options,
title=title,
next_enabled=False)
return slayout.get_seed(
), slayout.is_bip39, slayout.is_ext, slayout.is_gold_wallet_import
else:
self.exec_layout(slayout, title, next_enabled=False)
if self.wallet_type in ['2-key', '3-key']:
return slayout.get_seed(
), slayout.is_gold_wallet_import, slayout.is_ext
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
@wizard_dialog
def add_xpub_dialog(self,
title,
message,
is_valid,
run_next,
allow_multi=False,
show_wif_help=False):
header_layout = QHBoxLayout()
label = WWLabel(message)
label.setMinimumWidth(400)
header_layout.addWidget(label)
if show_wif_help:
header_layout.addWidget(InfoButton(get_wif_help_text()),
alignment=Qt.AlignRight)
return self.text_input(title, header_layout, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d" % index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.'
)
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _(
'Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options, on_import=True)
@wizard_dialog
def confirm_seed_dialog(self, run_next, test):
self.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.'
)
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text):
title = _("Your wallet's generated seed is:")
slayout = SeedLayout(seed=seed_text,
title=title,
msg=True,
options=['ext'])
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind, force_disable_encrypt_cb):
playout = PasswordLayout(
msg=msg,
kind=kind,
OK_button=self.next_button,
force_disable_encrypt_cb=force_disable_encrypt_cb)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.new_password(), playout.encrypt_cb.isChecked()
@wizard_dialog
def request_password(self, run_next, force_disable_encrypt_cb=False):
"""Request the user enter a new password and confirm it. Return
the password or None for no password."""
msg_enter_password = (
_("Choose a password to encrypt your wallet keys.") + '\n' +
_("Leave this field empty if you want to disable encryption."))
return self.pw_layout(msg_enter_password, PW_NEW,
force_disable_encrypt_cb)
@wizard_dialog
def request_storage_encryption(self, run_next):
msg_hw_storage_encryption = (_(
"Set wallet file encryption."
) + '\n' + _(
"Your wallet file does not contain secrets, mostly just metadata. "
) + _(
"It also contains your master public key that allows watching your addresses."
) + '\n\n' + _(
"Note: If you enable this setting, you will need your hardware device to open your wallet."
))
playout = PasswordLayoutForHW(msg_hw_storage_encryption)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.encrypt_cb.isChecked()
@wizard_dialog
def confirm_dialog(self, title, message, run_next):
self.confirm(message, title)
def confirm(self, message, title):
label = WWLabel(message)
vbox = QVBoxLayout()
vbox.addWidget(label)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self, **kwargs):
self.accept_signal.emit()
def waiting_dialog(self, task, msg, on_finished=None):
label = WWLabel(msg)
vbox = QVBoxLayout()
vbox.addSpacing(100)
label.setMinimumWidth(300)
label.setAlignment(Qt.AlignCenter)
vbox.addWidget(label)
self.set_layout(vbox, next_enabled=False)
self.back_button.setEnabled(False)
t = threading.Thread(target=task)
t.start()
while True:
t.join(1.0 / 60)
if t.is_alive():
self.refresh_gui()
else:
break
if on_finished:
on_finished()
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
id_ = self.compute_window_id(message, c_titles,
list(self.data.values()))
checked_index = LastChosenState.get_index(id_)
clayout = ChoicesLayout(message, c_titles, checked_index=checked_index)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
LastChosenState.set_state(id_, clayout.selected_index())
return action
@wizard_dialog
def choice_dialog_with_advanced_options(self, title, message, base_choices,
advanced_choices, run_next):
choices = base_choices + advanced_choices
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
id_ = self.compute_window_id(message, c_titles,
list(self.data.values()))
checked_index = LastChosenState.get_index(id_)
clayout = ChoicesLayout(message, c_titles, checked_index=checked_index)
if checked_index < len(base_choices):
clayout.show_index(len(base_choices))
self._show_advanced_text()
else:
clayout.show_index(len(choices))
self._hide_advanced_text()
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_advanced_layout(
layout=vbox,
default_show_function=lambda: clayout.show_index(len(base_choices)
),
advanced_show_function=lambda: clayout.show_index(len(choices)),
title=title,
)
action = c_values[clayout.selected_index()]
LastChosenState.set_state(id_, clayout.selected_index())
return action
@wizard_dialog
def get_recovery_pubkey(self, run_next, instant_key=None):
# todo move it to some global settings ?
web_generator_url = 'https://keygenerator.bitcoinvault.global/'
label = QLabel()
message = _(
'Please paste a Cancel Transaction Key. Use an existing one, if you are importing a wallet, '
'or generate a new one at')
message += f' <a href="{web_generator_url}">{web_generator_url}</a>'
label.setText(message)
label.setOpenExternalLinks(True)
label.setTextInteractionFlags(Qt.TextBrowserInteraction)
label.setWordWrap(True)
disallowed_keys = [instant_key] if instant_key else []
layout = InsertPubKeyDialog(self,
message_label=label,
disallowed_keys=disallowed_keys)
self.exec_layout(layout,
_('Cancel Transaction Key'),
next_enabled=False)
return layout.get_compressed_pubkey()
@wizard_dialog
def get_instant_pubkey(self, run_next, recovery_key=None):
# todo move it to some global settings ?
web_generator_url = 'https://keygenerator.bitcoinvault.global/'
label = QLabel()
message = _(
'Please paste a Fast Transaction Key. Use an existing one, if you are importing a wallet, '
'or generate a new one at')
message += f' <a href="{web_generator_url}">{web_generator_url}</a>.'
label.setText(message)
label.setOpenExternalLinks(True)
label.setTextInteractionFlags(Qt.TextBrowserInteraction)
label.setWordWrap(True)
disallowed_keys = [recovery_key] if recovery_key else []
layout = InsertPubKeyDialog(self,
message_label=label,
disallowed_keys=disallowed_keys)
self.exec_layout(layout, _('Fast Transaction Key'), next_enabled=False)
return layout.get_compressed_pubkey()
@wizard_dialog
def get_authenticator_pubkey(self, run_next, disallowed_key=None):
# todo move it to some global settings ?
label = QLabel()
message = _(
'Please paste authenticator public key exported from your Gold Wallet application'
)
label.setText(message)
label.setOpenExternalLinks(True)
label.setTextInteractionFlags(Qt.TextBrowserInteraction)
label.setWordWrap(True)
disallowed_keys = [disallowed_key] if disallowed_key else []
layout = InsertPubKeyDialog(self,
message_label=label,
disallowed_keys=disallowed_keys)
self.exec_layout(layout,
_('Gold Wallet authenticator public key'),
next_enabled=False)
return layout.get_compressed_pubkey()
@wizard_dialog
def display_2fa_pairing_qr(self, run_next, entropy: bytes):
title_label = QLabel()
title_message = _(
'Generate an authenticator on Gold Wallet and scan the following QR code: '
)
title_label.setText(title_message)
title_label.setWordWrap(True)
pin_label = QLabel()
pin_message = _('Confirmation PIN from Gold Wallet:')
pin_label.setText(pin_message)
qr_data = {
"name": os.path.basename(self.path),
"entropy": entropy,
}
layout = Qr2FaDialog(self,
title_label=title_label,
pin_label=pin_label,
qr_data=qr_data)
if self.wallet_type == '2-key':
title = _('Pair your 2-Key Vault wallet with Gold Wallet')
elif self.wallet_type == '3-key':
title = _('Pair your 3-Key Vault wallet with Gold Wallet')
else:
raise ValueError(f'Wrong wallet type {self.wallet_type}')
self.exec_layout(layout, title, next_enabled=False)
return layout.get_pubkey()
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def choice_and_line_dialog(self,
title: str,
message1: str,
choices: List[Tuple[str, str, str]],
message2: str,
test_text: Callable[[str], int],
run_next,
default_choice_idx: int = 0) -> Tuple[str, str]:
vbox = QVBoxLayout()
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
c_default_text = [x[2] for x in choices]
def on_choice_click(clayout):
idx = clayout.selected_index()
line.setText(c_default_text[idx])
clayout = ChoicesLayout(message1,
c_titles,
on_choice_click,
checked_index=default_choice_idx)
vbox.addLayout(clayout.layout())
vbox.addSpacing(50)
vbox.addWidget(WWLabel(message2))
line = QLineEdit()
def on_text_change(text):
self.next_button.setEnabled(test_text(text))
line.textEdited.connect(on_text_change)
on_choice_click(clayout) # set default text for "line"
vbox.addWidget(line)
self.exec_layout(vbox, title)
choice = c_values[clayout.selected_index()]
return str(line.text()), choice
@wizard_dialog
def line_dialog(self,
run_next,
title,
message,
default,
test,
warning='',
presets=(),
warn_issue4566=False):
vbox = QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
if warn_issue4566:
text_whitespace_normalised = ' '.join(text.split())
warn_issue4566_label.setVisible(
text != text_whitespace_normalised)
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
warn_issue4566_label = WWLabel(
_("Warning") + ": " +
_("You have multiple consecutive whitespaces or leading/trailing "
"whitespaces in your passphrase.") + " " +
_("This is discouraged.") + " " +
_("Due to a bug, old versions of Electrum will NOT be creating the "
"same wallet as newer versions or other software."))
warn_issue4566_label.setVisible(False)
vbox.addWidget(warn_issue4566_label)
for preset in presets:
button = QPushButton(preset[0])
button.clicked.connect(
lambda __, text=preset[1]: line.setText(text))
button.setMinimumWidth(150)
hbox = QHBoxLayout()
hbox.addWidget(button, alignment=Qt.AlignCenter)
vbox.addLayout(hbox)
self.exec_layout(vbox, title, next_enabled=test(default))
return line.text()
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QVBoxLayout()
layout = SeedLayout(xpub, title=msg, icon=False, for_seed_words=False)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network):
message = _("Electrum communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfill the same purpose only differing in "
"hardware. In most cases you simply want to let Electrum "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_('Cancel'))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
if r == 1:
nlayout = NetworkChoiceLayout(network, self.config, wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
else:
network.auto_connect = True
self.config.set_key('auto_connect', True, True)
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QSlider(Qt.Horizontal, self)
n_edit = QSlider(Qt.Horizontal, self)
n_edit.setMinimum(2)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QLabel()
m_label = QLabel()
grid = QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require {0} signatures').format(m))
cw.set_m(m)
def on_n(n):
n_label.setText(_('From {0} cosigners').format(n))
cw.set_n(n)
m_edit.setMaximum(n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
on_n(2)
on_m(2)
vbox = QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(
WWLabel(
_("Choose the number of signatures needed to unlock funds in your wallet:"
)))
vbox.addLayout(grid)
self.exec_layout(vbox, _("Multi-Signature Standard"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)
).setSectionResizeMode(QHeaderView.Stretch)
# self.home_page.ui.members_list.resizeRowsToContents()
self.all_payments.ui.all_payments_view.setSelectionBehavior(
QTableView.SelectRows)
self.all_payments.ui.all_payments_view.font().setPointSize(42)
self.all_payments.ui.all_payments_view.setSortingEnabled(True)
if __name__ == "__main__":
import sys, time
app = QApplication(sys.argv)
splash = MySplashScreen("images/giphy2.gif", Qt.Qt.WindowStaysOnTopHint)
# splash.setMask(splash_pix.mask())
# splash.raise_()
splash.show()
app.processEvents()
# this event loop is needed for dispatching of Qt events
initLoop = QEventLoop()
pool = Pool(processes=1)
pool.apply_async(longInitialization, [4],
callback=lambda exitCode: initLoop.exit(exitCode))
initLoop.exec_()
a = TheApp()
a.home_page.show()
splash.finish(a)
app.exec_()
class Kiwoom(QAxWidget):
""" 싱글톤 패턴 적용 """
__instance = None
@classmethod
def __getInstance(cls):
return cls.__instance
@classmethod
def instance(cls, *args, **kwargs):
cls.__instance = cls(*args, **kwargs)
cls.instance = cls.__getInstance
return cls.__instance
def __init__(self, API_SERVER_URL, API_SERVER_PORT):
super().__init__()
self.API_SERVER_URL = API_SERVER_URL
self.API_SERVER_PORT = API_SERVER_PORT
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
self.connectState = 0 # 최초에는 미접속
# old process kill
# self.__killOldProcess()
# Loop 변수: 비동기 방식으로 동작되는 이벤트를 동기화
self.logingLoop = None
self.requestLoop = None
self.orderLoop = None
self.conditionLoop = None
# 서버구분
self.serverStatus = None
# 연속조회구분
self.isNext = 0
# logging 클래스
self.homepath = os.environ.get('userprofile')
self.logger = Logger(path=self.log_path, name="Kiwoom")
# API 요청 제한 관리 Queue (1초 5회, 1시간 1,000회)
self.requestDelayCheck = APIDelayCheck(logger=self.logger)
self.orderDelayCheck = APIDelayCheck(logger=self.logger)
# 서버에서 받은 메시지
self.msg = ""
# Event 처리
self.OnEventConnect.connect(self.eventConnect)
self.OnReceiveTrData.connect(self.eventReceiveTrData)
self.OnReceiveRealData.connect(self.eventReceiveRealData)
self.OnReceiveConditionVer.connect(self.eventReceiveConditionVer)
self.OnReceiveTrCondition.connect(self.eventReceiveTrCondition)
self.OnReceiveRealCondition.connect(self.eventReceiveRealCondition)
self.OnReceiveChejanData.connect(self.eventReceiveChejanData)
self.OnReceiveMsg.connect(self.eventReceiveMsg)
@property
def log_path(self):
path = os.path.join(self.homepath, '.kiwoom_log')
if not os.path.exists(path):
os.mkdir(path)
return path
@property
def order_log_path(self):
path = os.path.join(self.homepath, '.kiwoom_order_log')
if not os.path.exists(path):
os.mkdir(path)
return path
###############################################################
################### Send log to api server ####################
###############################################################
def send_log(self, log):
response = requests.post(
'http://{}:{}/kiwoom_info/receive_response'.format(
self.API_SERVER_URL, self.API_SERVER_PORT),
json=json.dumps(log, ensure_ascii=False),
headers=headers)
if response.status_code == 200:
print('Send log success')
else:
print('Send log failed = {}'.format('!'))
###############################################################
################### 이벤트 발생 시 메서드 #####################
###############################################################
def eventConnect(self, returnCode):
""" 통신 연결 상태 변경시 이벤트
returnCode가 0이면 로그인 성공
그 외에는 ReturnCode 클래스 참조.
Parameters
----------
returnCode: int
0이면 로그인 성공, 이외에는 로그인 실패
"""
if returnCode == 0:
msg = "{} Connection Successful".format(dt.now())
self.connectState = 1
else:
errorName = getattr(ReturnCode, "CAUSE").get(returnCode)
msg = "{} Connection Failed : {}".format(dt.now(), errorName)
self.connectState = 0
self.logger.debug(msg)
try:
self.loginLoop.exit()
print('loginLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
return
def eventReceiveMsg(self, scrNo, rqName, trCode, msg):
""" 수신 메시지 이벤트
서버로 어떤 요청을 했을 때(로그인, 주문, 조회 등),
그 요청에 대한 처리내용을 전달해준다.
Parameters
----------
scrNo: str
화면번호(4자리, 사용자 정의,
서버에 조회나 주문을 요청할 때 이 요청을 구별하기 위한 키값)
rqName: str
TR 요청명(사용자 정의)
trCode: str
msg: str
서버로 부터의 메시지
"""
if hasattr(self, "orderResponse"):
self.orderResponse.update({"msg": msg})
self.logger.debug(msg)
def eventReceiveTrData(self, scrNo, rqName, trCode, recordName, inquiry,
**kwargs):
"""
TR 수신 이벤트시 실행되는 매서드
조회요청 응답을 받거나 조회데이터를 수신했을 때 호출됩니다.
rqName과 trCode는 getCommData() 메소드의 매개변수와 매핑되는 값 입니다.
요청한 TR에 대해 수신된 데이터는 self.{trCode}에 저장됩니다.
수신된 데이터는 모두 str 타입으로 사용자가 원하는 형태로 2차 가공이 필요합니다.
trCode 및 rqName과 관련된 자세한 내용은 OPEN API+ 개발가이드 및
KOA StudioSA를 참고하시기 바랍니다.
Parameters
----------
scrNo: str
화면번호(4자리)
rqName: str
TR 요청명(commRqData() 메소드 호출시 사용된 rqName)
trCode: str
recordName: str
inquiry: str
조회("0" or "": 남은 데이터 없음, '2': 남은 데이터 있음)
"""
# 주문 이벤트인 경우
if "ORD" in trCode:
# 주문번호 획득, 주문번호가 존재하면 주문 성공
orderNo = self.getCommData(trCode, "", 0, "주문번호")
self.orderResponse.update({"orderNo": orderNo})
try:
self.orderLoop.exit()
print('orderLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
pass
return
# 취소 이벤트인 경우
if "KOA_NORMAL_KP_CANCEL" in trCode:
# 주문번호 획득, 주문번호가 존재하면 주문 성공
orderNo = self.getCommData(trCode, "", 0, "주문번호")
self.orderResponse.update({"orderNo": orderNo})
try:
self.orderLoop.exit()
print('orderLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
pass
return
# 수정 이벤트인 경우
if "KOA_NORMAL_KP_MODIFY" in trCode:
# 주문번호 획득, 주문번호가 존재하면 주문 성공
orderNo = self.getCommData(trCode, "", 0, "주문번호")
self.orderResponse.update({"orderNo": orderNo})
try:
self.orderLoop.exit()
print('orderLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
pass
return
# TR 이벤트인 경우, orderResponse를 삭제
if hasattr(self, "orderResponse"):
delattr(self, "orderResponse")
# TR Data 수신
if trCode == "OPTKWFID":
data = self.__getOPTKWFID(trCode, rqName)
else:
data = self.__getData(trCode, rqName)
setattr(self, trCode, data)
self.isNext = 0 if ((inquiry == "0") or
(inquiry == "")) else 2 # 추가조회 여부
# TR loop 탈출
try:
self.requestLoop.exit()
print('requestLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
pass
# TR 이벤트 logging
# eventDetail = {
# "TIME": dt.now().strftime("%Y-%m-%d %H:%M:%S.%f"),
# "BASC_DT": dt.now().strftime("%Y-%m-%d"),
# "EVENT": "eventReceiveTrData",
# "REQUEST_NAME": rqName,
# "TR_CODE": trCode,
# }
# print(eventDetail)
def eventReceiveRealData(self, code, realType, realData):
""" 설정된 종목에 대한 실시간 정보 수신시 발생하는 이벤트
Parameters
----------
code: str
종목코드
realType: str
fid의 갯수
realData: str
fidList 구분은 \t 이다.
"""
print('code = ', code)
print('realType = ', realType)
if realType in RealType.FIDS:
fidList = getattr(RealType, realType)
resultDict = {}
for (k, v) in fidList.items():
resultDict[v] = self.getCommRealData(realType, k)
print('realType = {}, return = {}'.format(realType, resultDict))
resultDict["BASC_DT"] = dt.now().strftime("%Y-%m-%d")
self.send_log({
'log_type': 'realtime',
'code': code,
'realType': realType,
'log': resultDict
})
# print('log response in realtype = ', response)
def eventReceiveChejanData(self, gubun, itemCnt, fidList):
""" 주문 접수/확인 수신시 이벤트
주문요청후 주문접수, 체결통보, 잔고통보를 수신할 때 마다 호출됩니다.
Parameters
----------
gubun: str
체결구분('0': 주문접수/주문체결, '1': 잔고통보, '3': 특이신호)
itemCnt: int
fid의 갯수
fidList: str
fidList 구분은 ;(세미콜론) 이다.
"""
if gubun != '0': # 주문접수/주문체결이 아니면 logging 안함
return
print('fid List in receive chejan data = ', fidList)
orderStatus = self.getChejanData(
'913').strip() # 주문상태 "접수" or "체결" or "확인"
print('orderStatus in chejan data = ', orderStatus)
if orderStatus == '접수':
fidDict = getattr(FidList, 'SUBMITTED')
elif orderStatus == '체결':
fidDict = getattr(FidList, 'EXECUTED')
elif orderStatus == '확인': #주문취소
fidDict = getattr(FidList, 'CANCELLED')
else:
fidDict = getattr(FidList, 'ALL')
print('fid dict for chejan data = ', fidDict)
resultDict = {"BASC_DT": dt.now().strftime("%Y-%m-%d")}
fids = fidList.split(";")
for fid in fids:
fidName = fidDict.get(fid)
if fidName is None:
continue
data = self.getChejanData(fid).strip()
resultDict[fidName] = data
self.send_log({'log_type': 'chejan', 'log': resultDict})
###############################################################
#################### 로그인 관련 메서드 ######################
###############################################################
def commConnect(self):
""" 로그인 시도 """
if not self.connectState:
self.dynamicCall("CommConnect()")
self.loginLoop = QEventLoop()
self.loginLoop.exec_() # eventConnect에서 loop를 종료
def commTerminate(self):
""" 로그인 시도 """
if not self.connectState:
self.dynamicCall("CommTerminate()")
self.loginLoop = QEventLoop()
self.loginLoop.exec_() # eventConnect에서 loop를 종료
def getConnectionState(self):
""" 현재 접속상태를 반환합니다.
Returns
----------
int
0(미연결), 1(연결)
"""
return self.dynamicCall("GetConnectState()")
@property
def accNo(self):
return self.getLoginInfo("ACCNO").rstrip(";")
def getLoginInfo(self, tag):
""" 사용자의 tag에 해당하는 정보를 반환한다.
tag에 올 수 있는 값은 아래와 같다.
ACCOUNT_CNT, ACCNO, USER_ID, USER_NAME, GetServerGubun
Parameters
----------
tag: str
isConnectState: bool
접속상태을 확인할 필요가 없는 경우 True로 설정.
Returns
-----------
info : str
입력한 tag에 대응하는 정보
"""
if not self.connectState: # 1: 연결, 0: 미연결
raise KiwoomConnectError()
tags = [
"ACCOUNT_CNT", "ACCNO", "USER_ID", "USER_NAME", "GetServerGubun"
]
if tag not in tags:
raise ParameterValueError()
if tag == "GetServerGubun":
info = self.getServerGubun()
else:
info = self.dynamicCall('GetLoginInfo("{}")'.format(tag))
return info
def getServerGubun(self):
""" 서버구분 정보를 반환한다.
Returns
----------
server_status: str
"1": 모의투자서버 else: 실서버
"""
serverStatus = self.dynamicCall("KOA_Functions(QString, QString)",
"GetServerGubun", "")
return serverStatus
###############################################################
################### TR(조회) 관련 메서드 #####################
## 시세조회, 관심종목 조회, 조건검색 등 합산 조회수 1초 5회 제한 ##
###############################################################
# 점검방법은 CommRqData()함수와 CommKwRqData()함수를 이용한 조회횟수를
# 합산하는 것으로 연속조회 역시 CommRqData()를 이용므로 합산됩니다.
def setInputValue(self, key, value):
""" TR 전송에 필요한 값을 설정한다.
Parameters
----------
key: str
TR에 명시된 input 이름, ex) 계좌번호, 종목코드
value: str
key에 해당하는 값, ex) 88231524, 005930
"""
if not isinstance(key, str):
key = str(key)
if not isinstance(value, str):
value = str(value)
if (key == "계좌번호") and (value not in self.accNo.split(';')):
raise KiwoomProcessingError("ERROR: Invalid 계좌번호")
if (key == "종목코드") and (value not in self.codes):
raise KiwoomProcessingError("ERROR: Invalid 종목코드")
self.dynamicCall("SetInputValue(QString, QString)", key, value)
def commRqData(self, rqName, trCode, inquiry, scrNo):
""" 키움서버에 TR 요청을 한다.
요청한 데이터는 데이터 수신 이벤트 발생 시 eventReceiveTrData 매서드에서 처리
1초에 5회 제한
Parameters
----------
rqName: str
TR 요청명(사용자 정의)
trCode: str
inquiry: int
조회(0: 조회, 2: 남은 데이터 이어서 요청)
scrNo: str
화면번호(4자리)
Returns
----------
returnCode: str
0(정상), -200(시세과부하), -201(조회전문작성 에러)
"""
if not self.connectState:
raise KiwoomConnectError()
if not (isinstance(rqName, str) and isinstance(trCode, str)
and isinstance(inquiry, int) and isinstance(scrNo, str)):
raise ParameterTypeError()
# API 제한 확인
self.requestDelayCheck.checkDelay()
returnCode = self.dynamicCall(
"CommRqData(QString, QString, int, QString)",
rqName,
trCode,
inquiry,
scrNo,
)
if returnCode != 0: # 0이외엔 실패
self.logger.error(
"{} commRqData {} Request Failed!, CAUSE: {}".format(
dt.now(), rqName,
getattr(ReturnCode, "CAUSE").get(returnCode)))
raise KiwoomProcessingError()
# 루프 생성: eventReceiveTrData() 메서드에서 루프를 종료시킨다.
self.logger.debug("{} commRqData {}".format(dt.now(), rqName))
self.requestLoop = QEventLoop()
self.requestLoop.exec_()
print('requestLoop entered')
def getRepeatCnt(self, trCode, rqName):
""" 서버로 부터 전달받은 데이터의 갯수를 리턴합니다.(멀티데이터의 갯수)
receiveTrData() 이벤트 메서드가 호출될 때, 그 안에서 사용해야 합니다.
키움 OpenApi+에서는 데이터를 싱글데이터와 멀티데이터로 구분합니다.
싱글데이터란, 서버로 부터 전달받은 데이터 내에서, 중복되는 키(항목이름)가
하나도 없을 경우. 예를들면, 데이터가 '종목코드', '종목명', '상장일',
'상장주식수' 처럼 키(항목이름)가 중복되지 않는 경우를 말합니다.
반면 멀티데이터란, 서버로 부터 전달받은 데이터 내에서, 일정 간격으로
키(항목이름)가 반복될 경우를 말합니다. 예를들면, 10일간의 일봉데이터를
요청할 경우 '종목코드', '일자', '시가', '고가', '저가' 이러한 항목이
10번 반복되는 경우입니다. 이러한 멀티데이터의 경우 반복 횟수(=데이터의 갯수)
만큼, 루프를 돌면서 처리하기 위해 이 메서드를 이용하여 멀티데이터의 갯수를
얻을 수 있습니다.
차트조회는 한번에 최대 900개 데이터를 수신할 수 있습니다.
Parameters
----------
trCode: str
rqName: str
Returns
----------
cnt : int
서버에서 전달받은 데이터 갯수
"""
if not (isinstance(trCode, str) and isinstance(rqName, str)):
raise ParameterTypeError()
cnt = self.dynamicCall("GetRepeatCnt(QString, QString)", trCode,
rqName)
return cnt
def getCommData(self, trCode, rqName, index, key):
""" 데이터 획득 메서드
evnetReceiveTrData() 이벤트 메서드가 호출될 때, 그 안에서
조회데이터를 얻어오는 메서드입니다. 이 함수는 반드시 OnReceiveTRData()
이벤트가 호출될때 그 안에서 사용해야 합니다.
싱글데이터는 index=0
멀티데이터는 getRepeatCnt 매서드로 데이터 수를 확인한 후,
loop문으로 index를 1씩 늘리며 접근
Parameters
-----------
trCode: str
rqName: str
TR 요청명(commRqData() 메소드 호출시 사용된 rqName)
index: int
key: str
수신 데이터에서 얻고자 하는 값의 키(출력항목이름)
Returns
----------
data: str
"""
if not (isinstance(trCode, str) and isinstance(rqName, str)
and isinstance(index, int) and isinstance(key, str)):
raise ParameterTypeError()
data = self.dynamicCall("GetCommData(QString, QString, int, QString)",
trCode, rqName, index, key)
return data.strip()
def getCommDataEx(self, trCode, multiDataName):
""" 멀티 데이터 획득
조회 수신데이터 크기가 큰 차트데이터를 한번에 가져올 목적으로 만든 전용함수입니다.
receiveTrData() 이벤트 메서드가 호출될 때, 그 안에서 사용해야 합니다.
Parameters
----------
trCode: str
multiDataName: str
KOA에 명시된 멀티데이터명
Returns
----------
data: list
중첩리스트
"""
if not (isinstance(trCode, str) and isinstance(multiDataName, str)):
raise ParameterTypeError()
data = self.dynamicCall("GetCommDataEx(QString, QString)", trCode,
multiDataName)
return data
def commKwRqData(self,
arrCode,
next,
codeCount,
rqName,
scrNo,
typeFlag=0):
""" 복수종목조회 메서드(관심종목조회 메서드라고도 함).
이 메서드는 setInputValue() 메서드를 이용하여, 사전에 필요한 값을 지정하지 않는다.
단지, 메서드의 매개변수에서 직접 종목코드를 지정하여 호출한다.
데이터 수신은 receiveTrData() 이벤트에서 아래 명시한 항목들을 1회 수신하며,
이후 receiveRealData() 이벤트를 통해 실시간 데이터를 얻을 수 있다.
복수종목조회 TR 코드는 OPTKWFID 이며, 요청 성공시 아래 항목들의 정보를 얻을 수 있다.
[종목코드, 종목명, 현재가, 기준가, 전일대비, 전일대비기호, 등락율, 거래량, 거래대금,
체결량, 체결강도, 전일거래량대비, 매도호가, 매수호가, 매도1~5차호가, 매수1~5차호가,
상한가, 하한가, 시가, 고가, 저가, 종가, 체결시간, 예상체결가, 예상체결량, 자본금,
액면가, 시가총액, 주식수, 호가시간, 일자, 우선매도잔량, 우선매수잔량,우선매도건수,
우선매수건수, 총매도잔량, 총매수잔량, 총매도건수, 총매수건수, 패리티, 기어링, 손익분기,
잔본지지, ELW행사가, 전환비율, ELW만기일, 미결제약정, 미결제전일대비, 이론가,
내재변동성, 델타, 감마, 쎄타, 베가, 로]
1초에 5회 제한
Parameters
----------
arrCode: str
종목코드, 세미콜론(;)으로 구분, 한번에 100종목까지 조회가능
next: int
(0: 조회, 1: 남은 데이터 이어서 조회)
기존 API 문서는 boolean type
codeCount: int
codes에 지정한 종목의 갯수.
rqName: str
scrNo: str
typeFlag: int
주식과 선물옵션 구분(0: 주식, 3: 선물옵션),
기존 API 문서에서는 가운데 위치하지만, 맨 뒤로 이동시켰음
return
----------
str
0(정상), -200(시세과부하), -201(조회전문작성 에러)
"""
if not self.connectState:
raise KiwoomConnectError()
if not (isinstance(arrCode, str) and isinstance(next, int)
and isinstance(codeCount, int) and isinstance(rqName, str)
and isinstance(scrNo, str) and isinstance(typeFlag, int)):
raise ParameterTypeError()
# API 제한 확인
self.requestDelayCheck.checkDelay()
returnCode = self.dynamicCall(
"CommKwRqData(QString, QBoolean, int, int, QString, QString)",
arrCode,
next,
codeCount,
typeFlag,
rqName,
scrNo,
)
if returnCode != ReturnCode.OP_ERR_NONE:
self.logger.error("{} commKwRqData {} Request Failed!".format(
dt.now(), rqName))
raise KiwoomProcessingError()
# logging
self.logger.debug("{} commKwRqData {}".format(dt.now(), rqName))
# eventReceiveTrData()에서 loop 종료 or timeout
self.requestLoop = QEventLoop()
QTimer.singleShot(1000, self.requestLoop.exit) # timout in 1000 ms
self.requestLoop.exec_()
print('requestLoop entered')
# strRealType – 실시간 구분
# nFid – 실시간 아이템
# Ex) 현재가출력 - openApi.GetCommRealData(“주식시세”, 10);
# 참고)실시간 현재가는 주식시세, 주식체결 등 다른 실시간타입(RealType)으로도 수신가능
# https://programtalk.com/vs2/python/11670/QWebview-plus/plus/kiwoom.py/
def getCommRealData(self, realType, fid):
if not (isinstance(realType, str) and isinstance(fid, str)):
raise ParameterTypeError()
data = self.dynamicCall("GetCommRealData(QString, int)", realType, fid)
return data.strip()
###############################################################
################### 조건식 관련 메서드 #################
########################## 1초 5회 제한 ########################
###############################################################
def getCondition(self):
returnCode = self.dynamicCall("GetConditionLoad()")
if returnCode != 1:
self.logger.error(
"{} GetConditionLoad {} Request Failed!, CAUSE: {}".format(
dt.now(), conditionName,
getattr(ReturnCode, "CAUSE").get(returnCode)))
raise KiwoomProcessingError()
# 루프 생성: eventReceiveTrData() 메서드에서 루프를 종료시킨다.
self.logger.debug("{} condition name list requested.".format(
dt.now()))
self.conditionLoop = QEventLoop()
self.conditionLoop.exec_()
print('conditionLoop entered')
def eventReceiveConditionVer(self, IRet):
print('on receive condition ver = ', IRet)
if IRet != 1:
self.logger.error(
"{} sendCondition {} Request Failed!, CAUSE: {}".format(
dt.now(), conditionName,
getattr(ReturnCode, "CAUSE").get(returnCode)))
raise KiwoomProcessingError()
print('Successfully got condition version!')
try:
self.conditionLoop.exit()
print('conditionLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
pass
def getConditionList(self):
return self.dynamicCall("GetConditionNameList()")
def getCompByCondition(self, scrNo, conditionName, nIndex, nSearch):
self.sendCondition(scrNo, conditionName, nIndex, nSearch)
return getattr(self, conditionName)
def sendCondition(self, scrNo, conditionName, nIndex, nSearch):
'''
# Params
BSTR strScrNo, // 화면번호
BSTR strConditionName, // 조건식 이름
int nIndex, // 조건명 인덱스
int nSearch // 조회구분, 0:조건검색, 1:실시간 조건검색
'''
if not isinstance(scrNo, str):
strScrNo = str(scrNo)
if not isinstance(conditionName, str):
conditionName = str(conditionName)
if not isinstance(nIndex, int):
nIndex = int(nIndex)
if not isinstance(nSearch, int):
nSearch = int(nSearch)
returnCode = self.dynamicCall(
"SendCondition(QString, QString, int, int)", scrNo, conditionName,
nIndex, nSearch)
print('returnCode for cond = {}'.format(returnCode))
if returnCode != 1: # 1이외엔 실패
self.logger.error(
"{} sendCondition {} Request Failed!, CAUSE: {}".format(
dt.now(), conditionName,
getattr(ReturnCode, "CAUSE").get(returnCode)))
raise KiwoomProcessingError()
self.logger.info("{} sendCondition {} Realtime: {}".format(
dt.now(), conditionName, nSearch))
# 루프 생성: eventReceiveTrCondition() 메서드에서 루프를 종료시킨다.
self.conditionLoop = QEventLoop()
self.conditionLoop.exec_()
# 실시간 조건검색을 중지합니다.
# ※ 화면당 실시간 조건검색은 최대 10개로 제한되어 있어서 더 이상 실시간 조건검색을 원하지 않는 조건은 중지해야만 카운트 되지 않습니다.
def sendConditionStop(self, scrNo, conditionName, index):
self.dynamicCall("SendConditionStop(QString, QString, int)", scrNo,
conditionName, index)
def eventReceiveTrCondition(self, scrNo, codeList, conditionName, nIndex,
nNext):
setattr(self, conditionName,
[x for x in codeList.split(';') if len(x) > 0])
try:
self.conditionLoop.exit()
print('conditionLoop Exited')
except AttributeError as e:
print('Attribute error after tr receive = {}', e)
return
# https://devshj.tistory.com/14
def eventReceiveRealCondition(
self, code, type, conditionName,
conditionIndex): # 조건검색 실시간 조회시 반환되는 값을 받는 함수
'''실시간 편입/이탈 종목이 발생될 때마다 호출됩니다.'''
print("receive_real_condition strCode: " + str(code) + ", Type: " +
str(type) + ", ConditionName: " + str(conditionName) +
", ConditionIndex: " + str(conditionIndex))
try:
print("receive_real_condition strCode: " + str(code) + ", Type: " +
str(type) + ", ConditionName: " + str(conditionName) +
", ConditionIndex: " + str(conditionIndex))
logDateTime = dt.today().strftime(
"%Y-%m-%d %H:%M:%S") # 화면에 노출할 날짜를 만듬 (YYYY-mm-dd HH:MM:SS 형태)
codeName = self.dynamicCall("GetMasterCodeName(QString)",
[code]).strip() # 종목 코드로 종목 이름을 가져옴
if str(type) == "I": # 편입 종목이라면
print(
str(logDateTime) + " 편입 신호 : " + str(code) + ", " +
str(codeName)) # 트레이딩 화면 내역에 로그를 남김
elif str(type) == "D": # 이탈 종목이라면
print(
str(logDateTime) + " 이탈 신호 : " + str(code) + ", " +
str(codeName)) # 트레이딩 화면 내역에 로그를 남김
resultDict = {
'code': code,
'name': codeName,
'type': type,
'conditionName': conditionName,
'conditionIndex': conditionIndex
}
self.send_log({
'log_type': 'realtime_condition',
'log': resultDict
})
except Exception as e:
print('Realtime Condition error = {}'.format(e))
return
###############################################################
############### 실시간 조회와 관련된 메서드 #####################
# https://programtalk.com/vs2/python/11670/QWebview-plus/plus/kiwoom.py/
# 실시간 등록을 한다.
# strScreenNo : 화면번호
# strCodeList : 종목코드리스트(ex: 039490;005930;…)
# strFidList : FID번호(ex:9001;10;13;…)
# 9001 – 종목코드
# 10 - 현재가
# 13 - 누적거래량
# strOptType : 타입(“0”, “1”)
# 타입 “0”은 항상 마지막에 등록한 종목들만 실시간등록이 됩니다.
# 타입 “1”은 이전에 실시간 등록한 종목들과 함께 실시간을 받고 싶은 종목을 추가로 등록할 때 사용합니다.
# ※ 종목, FID는 각각 한번에 실시간 등록 할 수 있는 개수는 100개 입니다.
def setRealReg(self, screenNo, codeList, fidList, optType):
return self.dynamicCall(
"SetRealReg(QString, QString, QString, QString)", screenNo,
codeList, fidList, optType)
# 리얼 시세를 끊는다.
# 화면 내 모든 리얼데이터 요청을 제거한다.
# 화면을 종료할 때 반드시 위 함수를 호출해야 한다.
# Ex) openApi.DisconnectRealData(“0101”);
def disconnectRealData(self, scnNo):
self.dynamicCall("DisconnectRealData(QString)", scnNo)
###############################################################
################### 주문과 잔고처리 관련 메서드 #################
########################## 1초 5회 제한 ########################
###############################################################
def sendOrder(
self,
rqName,
scrNo,
accNo,
orderType,
code,
qty,
price,
hogaType,
originOrderNo,
):
""" 주식 주문 메서드
sendOrder() 메소드 실행시,
OnReceiveMsg, OnReceiveTrData, OnReceiveChejanData 이벤트가 발생한다.
이 중, 주문에 대한 결과 데이터를 얻기 위해서는 OnReceiveChejanData 이벤트를 통해서 처리한다.
OnReceiveTrData 이벤트를 통해서는 주문번호를 얻을 수 있는데,
주문후 OnReceiveTrData에서 주문번호가 ''공백으로 전달되면 주문접수 실패를 의미한다.
※ 시장가, 최유리지정가, 최우선지정가, 시장가IOC, 최유리IOC,시장가FOK,
최유리FOK, 장전시간외, 장후시간외 주문시
주문가격(price)을 0으로 입력.
Paramters
----------
rqName: str
주문 요청명(사용자 정의)
scrNo: str
화면번호(4자리)
accNo: str
계좌번호(10자리)
orderType: int
주문유형(1: 신규매수, 2: 신규매도, 3: 매수취소, 4: 매도취소, 5: 매수정정, 6: 매도정정)
code: str
종목코드
qty: int
주문수량
price: int
주문단가
hogaType: str
거래구분(00: 지정가, 03: 시장가, 05: 조건부지정가, 06: 최유리지정가, 그외에는 api 문서참조)
originOrderNo: str
원주문번호(신규주문에는 공백, 정정및 취소주문시 원주문번호를 입력합니다.)
"""
orderParams = {
"rqName": rqName,
"scrNo": scrNo,
"accNo": accNo,
"orderType": orderType,
"code": code,
"qty": qty,
"price": price,
"hogaType": hogaType,
"originOrderNo": originOrderNo,
}
# order response data
self.orderResponse = {
"time": dt.now().strftime("%Y-%m-%d %H:%M:%S.%f"),
"orderNo": "",
}
self.orderResponse.update(orderParams)
# server connection check
if not self.connectState:
msg = "Server not connected"
self.orderResponse.update({"msg": msg})
raise KiwoomConnectError(msg)
# Error: code not supported
if not code in self.codes:
msg = f"Code not supported: {code}"
self.orderResponse.update({"msg": msg})
raise KiwoomProcessingError("ERROR: sendOrder() : {}".format(msg))
# API 제한 확인
self.orderDelayCheck.checkDelay()
# 주문 전송
try:
returnCode = self.dynamicCall(
"SendOrder(QString, QString, QString, int, QString, int, int, QString, QString)",
list(orderParams.values()),
)
except Exception as msg:
self.orderResponse.update({"msg": msg})
raise KiwoomProcessingError("ERROR: sendOrder() : {}".format(msg))
if returnCode != 0:
msg = getattr(ReturnCode, "CAUSE").get(returnCode)
self.orderResponse.update({"msg": msg})
raise KiwoomProcessingError("ERROR: sendOrder() : {}".format(msg))
# eventReceiveTrData() 에서 루프종료 or timeout
self.orderLoop = QEventLoop()
#QTimer.singleShot(1000, self.orderLoop.exit) # timout in 1000 ms
self.orderLoop.exec_()
print('orderLoop entered')
def getChejanData(self, fid):
""" 주문접수, 주문체결, 잔고정보를 얻어오는 메서드
이 메서드는 receiveChejanData() 이벤트 메서드가 호출될 때
그 안에서 사용해야 합니다.
Parameters
===========================
fid: str
return
===========================
data: str
"""
if not isinstance(fid, str):
raise ParameterTypeError()
data = self.dynamicCall(f'GetChejanData("{fid}")')
return data
def __getData(self, trCode, rqName):
returnDict = {}
if getattr(TRKeys, trCode).get("멀티데이터", False):
returnDict["멀티데이터"] = self.__getMultiData(trCode, rqName)
if getattr(TRKeys, trCode).get("싱글데이터", False):
returnDict["싱글데이터"] = self.__getSingleData(trCode, rqName)
return returnDict
def __getSingleData(self, trCode, rqName):
data = {}
keyList = getattr(TRKeys, trCode).get("싱글데이터")
for key in keyList:
val = self.getCommData(trCode, rqName, 0, key)
if key.endswith("호가") or key in getattr(TRKeys, "NOSIGNKEY"):
val = removeSign(val)
data[key] = val
return data
def __getMultiData(self, trCode, rqName):
data = []
cnt = self.getRepeatCnt(trCode, rqName)
keyList = getattr(TRKeys, trCode).get("멀티데이터")
for i in range(cnt):
tmpDict = {}
for key in keyList:
val = self.getCommData(trCode, rqName, i, key)
if key.endswith("호가") or key in getattr(TRKeys, "NOSIGNKEY"):
val = removeSign(val)
tmpDict[key] = val
data.append(tmpDict)
return data
def __getOPTKWFID(self, trCode, rqName):
data = {}
tmpData = self.getCommDataEx(trCode, rqName)
keyList = getattr(TRKeys, trCode).get("멀티데이터")
for key, ls in zip(keyList, zip(*tmpData)):
if key.endswith("호가") or key in getattr(TRKeys, "NOSIGNKEY"):
ls = map(removeSign, ls)
data[key] = list(ls)
data = dictListToListDict(data) # dict of list to list of dict
data = {"멀티데이터": data}
return data
def __getCodeListByMarket(self, market):
"""시장 구분에 따른 종목코드의 목록을 List로 반환한다.
market에 올 수 있는 값은 아래와 같다.
{
'0': 장내,
'3': ELW,
'4': 뮤추얼펀드,
'5': 신주인수권,
'6': 리츠,
'8': ETF,
'9': 하이일드펀드,
'10': 코스닥,
'30': 제3시장
}
Parameters
----------
market: str
Returns
----------
codeList: list
조회한 시장에 소속된 종목 코드를 담은 list
"""
if not self.connectState:
raise KiwoomConnectError()
if not isinstance(market, str):
raise ParameterTypeError()
if market not in ["0", "3", "4", "5", "6", "8", "9", "10", "30"]:
raise ParameterValueError()
codes = self.dynamicCall('GetCodeListByMarket("{}")'.format(market))
return codes.split(";")
@property
def codes(self):
if not self.connectState:
raise KiwoomConnectError()
codes = self.__getCodeListByMarket("0") # KOSPI
codes += self.__getCodeListByMarket("10") # KOSDAQ
codes += self.__getCodeListByMarket("8") # ETF
return codes
"""
class PrepareNextDay(ParentKiwoom):
def __init__(self):
super().__init__()
self.logging.logger.info("ETF PrepareNextDay() class start.")
self.line.notification("ETF PrepareNextDay() class start.")
self.all_etc_info_event_loop = QEventLoop()
self.etf_info_event_loop = QEventLoop()
self.etf_day_info_event_loop = QEventLoop()
self.screen_all_etf_stock = "4000"
self.screen_etf_stock = "5000"
self.screen_etf_day_stock = "4050"
self.target_etf_stock_dict = {}
self.target_etf_day_info_dict = []
self.event_slots()
self.line.notification("ETF PREPARE AUTO TRADE START")
self.prepare_next_day()
def event_slots(self):
self.OnReceiveTrData.connect(self.trdata_slot)
self.OnReceiveMsg.connect(self.msg_slot)
def prepare_next_day(self):
self.logging.logger.info("다음날 위한 준비 시작")
self.file_delete()
self.get_all_etf_stock()
self.get_etf_stock_info()
self.get_etf_day_stock_info()
QTest.qWait(5000)
self.createTargetEtfStockFile()
self.line.notification("시스템 종료")
QTest.qWait(5000)
sys.exit()
def file_delete(self):
self.logging.logger.info("file_delete")
if os.path.isfile(self.target_etf_file_path):
os.remove(self.target_etf_file_path)
self.logging.logger.info("remove %s" % self.target_etf_file_path)
def get_all_etf_stock(self, sPrevNext="0"):
self.logging.logger.info("get_all_etf_stock")
self.dynamicCall("SetInputValue(QString, QString)", self.customType.TAXATION_TYPE, "0")
self.dynamicCall("SetInputValue(QString, QString)", self.customType.COMPARED_TO_NAV, "0")
self.dynamicCall("SetInputValue(QString, QString)", self.customType.MANAGER, "0000")
self.dynamicCall("CommRqData(QString, QString, int, QString)", self.customType.OPT40004, "opt40004", sPrevNext, self.screen_all_etf_stock)
self.all_etc_info_event_loop.exec_()
def trdata_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
if sRQName == self.customType.OPT40004:
self.trdata_slot_opt40004(sScrNo, sRQName, sTrCode, sRecordName, sPrevNext)
elif sRQName == self.customType.OPT10001:
self.trdata_slot_opt10001(sScrNo, sRQName, sTrCode, sRecordName, sPrevNext)
elif sRQName == self.customType.OPT40003:
self.trdata_slot_opt40003(sScrNo, sRQName, sTrCode, sRecordName, sPrevNext)
def trdata_slot_opt40003(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
rows = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode, sRQName)
compareLastDayList = []
for i in range(rows):
compareLastDay = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, i, self.customType.THE_DAY_BEFORE)
compareLastDay = compareLastDay.strip()
if int(compareLastDay) > 0:
compareLastDayList.append(int(compareLastDay))
avgValue = round(int(np.mean(compareLastDayList)))
maxValue = np.max(compareLastDayList)
minValue = np.min(compareLastDayList)
for code in self.target_etf_day_info_dict:
self.target_etf_stock_dict[code].update({self.customType.THE_DAY_BEFORE_AVG: avgValue})
self.target_etf_stock_dict[code].update({self.customType.THE_DAY_BEFORE_MAX: maxValue})
self.target_etf_stock_dict[code].update({self.customType.THE_DAY_BEFORE_MIN: minValue})
self.logging.logger.info(self.logType.OPT40003_STATUS_LOG % (code, avgValue, maxValue, minValue))
del self.target_etf_day_info_dict[:]
self.etf_day_info_event_loop.exit()
def trdata_slot_opt40004(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
rows = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode, sRQName)
for i in range(rows):
volume = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, i, self.customType.VOLUME)
volume = volume.strip()
code = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, i, self.customType.STOCK_CODE)
code = code.strip()
last_price = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, i, self.customType.LAST_PRICE)
last_price = last_price.strip()
if int(volume) >= 50000 and int(last_price) <= 50000:
code_nm = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, i, self.customType.STOCK_NAME)
code_nm = code_nm.strip()
if code in self.target_etf_stock_dict:
pass
else:
self.target_etf_stock_dict[code] = {}
if sPrevNext == "2": # 다음페이지 존재
self.get_all_etf_stock(sPrevNext="2")
else:
self.stop_screen_cancel(self.screen_all_etf_stock)
self.all_etc_info_event_loop.exit()
def trdata_slot_opt10001(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
code = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.STOCK_CODE)
code = code.strip()
code_nm = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.STOCK_NAME)
highest_stock_price = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.HIGHEST_PRICE)
lowest_stock_price = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.LOWEST_PRICE)
last_stock_price = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.CURRENT_PRICE)
change_price = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.THE_DAY_BEFORE)
market_cap = self.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, self.customType.MARTKET_CAP)
market_cap = market_cap.strip()
self.logging.logger.info(self.logType.OPT10001_STATUS_LOG % (
code, highest_stock_price.strip(), lowest_stock_price.strip(), last_stock_price.strip(), change_price.strip(), market_cap)
)
self.target_etf_stock_dict[code].update({self.customType.STOCK_NAME: code_nm})
self.target_etf_stock_dict[code].update({self.customType.LAST_DAY_HIGHEST_PRICE: abs(int(highest_stock_price.strip()))})
self.target_etf_stock_dict[code].update({self.customType.LAST_DAY_LOWEST_PRICE: abs(int(lowest_stock_price.strip()))})
self.target_etf_stock_dict[code].update({self.customType.LAST_DAY_LAST_PRICE: abs(int(last_stock_price.strip()))})
self.target_etf_stock_dict[code].update({self.customType.MARTKET_CAP: int(market_cap)})
self.etf_info_event_loop.exit()
def isTragetEtfStock(self, value):
return int(value[self.customType.MARTKET_CAP]) >= 65
def createTargetEtfStockFile(self):
self.logging.logger.info("create_target_etf_stock_file")
for sCode in self.target_etf_stock_dict.keys():
value = self.target_etf_stock_dict[sCode]
if self.isTragetEtfStock(value):
f = open(self.target_etf_file_path, "a", encoding="utf8")
f.write("%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s\n" %
(sCode, value[self.customType.STOCK_NAME], value[self.customType.LAST_DAY_HIGHEST_PRICE],
value[self.customType.LAST_DAY_LOWEST_PRICE], value[self.customType.LAST_DAY_LAST_PRICE],
value[self.customType.THE_DAY_BEFORE_AVG], value[self.customType.THE_DAY_BEFORE_MAX],
value[self.customType.THE_DAY_BEFORE_MIN]))
f.close()
def get_etf_stock_info(self):
self.logging.logger.info("get_etf_stock_info")
for sCode in self.target_etf_stock_dict.keys():
QTest.qWait(4000)
self.dynamicCall("SetInputValue(QString, QString)", self.customType.STOCK_CODE, sCode)
self.dynamicCall("CommRqData(QString, QString, int, QString)", self.customType.OPT10001, "opt10001", 0, self.screen_etf_stock)
self.etf_info_event_loop.exec_()
def get_etf_day_stock_info(self):
self.logging.logger.info("get_etf_day_stock_info")
for sCode in self.target_etf_stock_dict.keys():
QTest.qWait(4000)
self.target_etf_day_info_dict.append(sCode)
self.dynamicCall("SetInputValue(QString, QString)", self.customType.STOCK_CODE, sCode)
self.dynamicCall("CommRqData(QString, QString, int, QString)", self.customType.OPT40003, "OPT40003", 0, self.screen_etf_day_stock)
self.etf_day_info_event_loop.exec_()
class Kiwoom(QAxWidget):
'''
키움 통신 클래스
=============
키움 API와 직접적인 통신을 진행한다.
Thread_safe함.
TODO
-----------
이벤트루프를 적용하여 get_tr_data 등의 메소드가 thread safe 해야함
'''
__global_eventloop: QEventLoop = None # 동기처리를 위한 전역 EventLoop
__tr_data_temp: Dict[str, Dict[str,
str]] = None # GetTrData의 결과가 임시적으로 담기는 곳
__condition_name_list: List[Tuple[str,
str]] = None # 조건식 이름과 인덱스가 임시적으로 담기는 곳
__condition_stock_list: List[str] = None # 조건식 필터링 결과가 임시적으로 담기는 곳
__tr_rq_single_data: Dict[str, str] = None # 사용자 요청 싱글데이터
__tr_rq_multi_data: Dict[str, str] = None # 사용자 요청 멀티데이터
__realtime_data_callback: Callable = None # 실시간 데이터 콜백
__server_msg_callback: Callable = None
__chejan_data_callback: Callable = None
def __init__(self):
super().__init__()
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
self.__global_eventloop = QEventLoop()
self.__reg_all_slot() # 이벤트 슬롯 등록
def do_login(self):
self.dynamicCall("CommConnect()")
self.__global_eventloop.exec_()
def get_tr_data(self, inputValue: Dict[str, str], trEnum: TrCode,
nPrevNext: int, sScrNo: str, rqSingleData: Dict[str, str],
rqMultiData: Dict[str, str]):
'''키움 API서버에 TR 데이터를 요청한다.
Parameters
----------
inputValue :
KOA Studio 기준으로, SetInputValue에 들어갈 값
trEnum : TrCode
KOA Studio 기준, Tr 코드
nPrevNext :
prevNext
sScrNo :
스크린 번호
rqSingleData :
받아오고자 하는 싱글데이터 목록
rqMultiData :
받아오고자 하는 멀티데이터 목록
Returns
-------
Dict[str, Dict[str, str]]
'''
self.__tr_rq_single_data = rqSingleData
self.__tr_rq_multi_data = rqMultiData
self._set_input_values(inputValue) # inputvalue 대입
r_value = self.dynamicCall(
"CommRqData(Qstring, QString, int, QString)", trEnum.value,
trEnum.name, nPrevNext, sScrNo)
print(ErrCode(r_value)) # dynamic call 반환 값 출력
self.__global_eventloop.exec_()
return self.__tr_data_temp
def get_condition_list(self):
'''키움 증권 HTS에 저장되어 있는 조건식 목록들을 반환
Returns
-------
(index, name)의 리스트를 반환한다.
'''
r_value = self.dynamicCall("GetConditionLoad()")
# print(ErrCode(r_value)) # dynamic call 반환 값 출력
self.__global_eventloop.exec_()
return self.__condition_name_list
def get_condition_stock(self, sScrNo: str, cond_name: str,
index: str) -> List[str]:
'''
해당 조건식을 만족하는 종목 코드를 리스트 형태로 반환
Parameters
----------
sScrNo : 스크린번호
cond_name : 조건식 이름
index : 조건식 인덱스 번호
Returns
-------
Stock 객체들의 리스트를 반환한다.
'''
r_value = self.dynamicCall("SendCondition(QString, QString, int, int)",
sScrNo, cond_name, index, 0)
print(ErrCode(r_value)) # dynamic call 반환 값 출력
self.__global_eventloop.exec_()
return self.__condition_stock_list
def set_realtime_reg(self, sScrNo: str, stockList: List[str],
realtimeDataList: List[RealTimeDataEnum]):
'''
realtime data를 받아올 수 있도록 레지스터링 한다.
Parameters
----------
stockList : Stock 객체가 list 형태로 들어와야함
realTimeDataList : RealTimeDataEnum 객체가 list 형태로 들어와야함
'''
if self.__realtime_data_callback is None:
raise RuntimeError("realtime data callback 함수가 설정되지 않았습니다.")
fid_list = "".join("{};".format(fid.value)
for fid in realtimeDataList)[:-1] # fid str list
stock_code_list = "".join(
"{};".format(stock)
for stock in stockList)[:-1] # stock code str list
r_value = self.dynamicCall(
"SetRealReg(QString, QString, QString, QString)", sScrNo,
stock_code_list, fid_list, 1)
print(ErrCode(r_value))
def set_realtime_callback(self, callback: Callable):
'''
realtime data 이벤트가 발생했을 때, 이를 받아들일 콜백을 설정한다.
'''
self.__realtime_data_callback = callback
def set_server_msg_callback(self, callback: Callable):
'''서버로 부터 메시지가 도착했을 때, 이를 출력할 콜백을 설정한다.
'''
self.__server_msg_callback = callback
def set_chejan_data_callback(self, callback: Callable):
self.__chejan_data_callback = callback
def send_order(self, sScrNo: str, sAccNo: str,
nOrderType: TrClassification, sCode: str, nQty: int,
nPrice: int, sHogaGb, sOrgOrderNo: str):
'''주식 주문
Parameters
----------
sScrNo :
스크린번호
sAccNo :
계좌번호
nOrderType :
주문 유형
sCode :
종목코드
nQty :
주문수량
nPrice :
주문가격
sHogaGb :
거래구분
sOrgOrderNo :
원주문번호
'''
self.dynamicCall(
"SendOrder(QString, QString, QString, QString, QString, QString, QString, QString, QString)",
TrCode.SEND_ORDER.value, sScrNo, sAccNo, nOrderType.value, sCode,
nQty, nPrice, sHogaGb, sOrgOrderNo)
def parse_realtime(
self, sCode: str, realtimeDataList: List[RealTimeDataEnum]
) -> Dict[RealTimeDataEnum, Any]:
'''
realtime_callback이 호출 되었을 경우에만 사용이 가능하다.
실시간 데이터 이벤트가 발생했을 때, FID 들을 파싱하여 가져오는 메소드
'''
realtime_data = {}
for realtime_enum in realtimeDataList:
realtime_data[realtime_enum] = self.dynamicCall(
"GetCommRealData(QString, QString)", sCode,
realtime_enum.value)
return realtime_data
def get_account_info(self):
account_list = self.dynamicCall("GetLoginInfo(QString)", "ACCNO")
account_num = account_list.split(';')[0]
return account_num
def _set_input_values(self, input_value: Dict[str, str]):
'''
SetInputVlaue() 동적 호출 iteration 용도
'''
for k, v in input_value.items():
self.dynamicCall("SetInputValue(QString, QString)", k, v)
def _tr_data_slot(self, sScrNo, sRQName, sTrCode, sPrevNext):
'''
CommRqData 처리용 슬롯
'''
_ = (sScrNo, sPrevNext) # warning avoid
n_record = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode,
sRQName) # tr데이터 중, 멀티데이터의 레코드 개수를 받아옴.
self.__tr_data_temp = {} # 이전에 저장되어 있던 임시 tr_data 삭제.
self.__tr_data_temp["single_data"] = {} # empty dict 선언
for s_data in self.__tr_rq_single_data:
self.__tr_data_temp["single_data"][s_data] = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, 0, s_data).replace(" ", "")
self.__tr_data_temp["multi_data"] = []
for i in range(n_record):
m_data_dict_temp = {} # 멀티데이터에서 레코드 하나에 담길 딕셔너리 선언
for m_data in self.__tr_rq_multi_data:
m_data_dict_temp[m_data] = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, m_data)
self.__tr_data_temp["multi_data"].append(m_data_dict_temp)
self.__global_eventloop.exit()
def _realtime_data_slot(self, sCode, sRealType, sRealData):
'''
realtime data 처리용 슬롯
'''
self.__realtime_data_callback(sCode, sRealType, sRealData)
def _login_slot(self, errNo):
print(ErrCode(errNo))
self.__global_eventloop.exit()
def _send_condition_slot(self, _sScrNo, sCodeList, _sCondName, _nIndex,
_nNext):
'''
SendCondition 처리용 슬롯
'''
self.__condition_stock_list = sCodeList[:-1].split(";")
self.__global_eventloop.exit()
def _condition_ver_slot(self, _lRet, _sMsg):
'''
GetConditionLoad 처리용 슬롯
'''
print(_sMsg)
self.__condition_name_list = [] # 이전에 저장되어 있던 조건식 들을 삭제한다.
condition_name_list = self.dynamicCall("GetConditionNameList()")
for cond_index_name in condition_name_list[:-1].split(";"):
self.__condition_name_list.append(cond_index_name.split("^"))
self.__global_eventloop.exit()
def _server_msg_slot(self, _sScrNo, sRQName, sTrCode, sMsg):
'''
서버 통신 응답 처리용 슬롯
'''
if self.__server_msg_callback is not None:
self.__server_msg_callback(TrCode(sRQName.replace(" ", ""), sMsg))
def _chejan_data_slot(self, sGubun, nItemCnt, sFidList):
'''체결 잔고데이터 이벤트 처리용 슬롯
'''
fid_list = sFidList[:-1].split(";")
chejan_fid_data = {}
for fid in fid_list:
chejan_fid_data[fid] = self.dynamicCall("GetChejanData(QString)",
fid)
if self.__chejan_data_callback is not None:
self.__chejan_data_callback(chejan_fid_data)
def __reg_all_slot(self):
'''
키움 API의 이벤트 슬롯을 전부 다 등록
'''
self.OnEventConnect.connect(self._login_slot) # 로그인 슬롯
self.OnReceiveTrData.connect(self._tr_data_slot) # Tr 데이터 슬롯
self.OnReceiveConditionVer.connect(
self._condition_ver_slot) # 조건식 데이터 슬롯
self.OnReceiveTrCondition.connect(
self._send_condition_slot) # 조건식 종목 슬롯
self.OnReceiveRealData.connect(self._realtime_data_slot) # 실시간 데이터 슬롯
self.OnReceiveMsg.connect(self._server_msg_slot)
self.OnReceiveChejanData.connect(self._chejan_data_slot)
class SphinxConverter(QObject):
"""This class converts Sphinx input to HTML. It is run in a separate
thread.
"""
# This signal clears the context of the log window.
logWindowClear = pyqtSignal()
# This signal emits messages for the log window.
logWindowText = pyqtSignal(
# A string to append to the log window.
str)
def __init__(self, parent):
super().__init__(parent)
# Use an additional thread to process Sphinx output.
self._ac = AsyncController('QThread', self)
self._ac.defaultPriority = QThread.LowPriority
self._SphinxInvocationCount = 1
def terminate(self):
# Free resources.
self._ac.terminate()
def convert(self, filePath):
# Run the builder.
errString = self._runHtmlBuilder()
# Look for the HTML output.
#
# Get an absolute path to the output path, which could be relative.
outputPath = core.config()['Sphinx']['OutputPath']
projectPath = core.config()['Sphinx']['ProjectPath']
if not os.path.isabs(outputPath):
outputPath = os.path.join(projectPath, outputPath)
# Create an htmlPath as OutputPath + remainder of filePath.
htmlPath = os.path.join(outputPath + filePath[len(projectPath):])
html_file_suffix = '.html'
try:
with codecs.open(os.path.join(projectPath, 'sphinx-enki-info.txt')) as f:
hfs = f.read()
# If the file is empty, then html_file_suffix wasn't defined
# or is None. In this case, use the default extension.
# Otherwise, use the extension read from the file.
if hfs:
html_file_suffix = hfs
except:
errString = "Warning: assuming .html extension. Use " + \
"the conf.py template to set the extension.\n" + errString
pass
# First place to look: file.html. For example, look for foo.py
# in foo.py.html.
htmlFile = htmlPath + html_file_suffix
# Second place to look: file without extension.html. For
# example, look for foo.html for foo.rst.
htmlFileAlter = os.path.splitext(htmlPath)[0] + html_file_suffix
# Check that the output file produced by Sphinx is newer than
# the source file it was built from.
if os.path.exists(htmlFile):
return _checkModificationTime(filePath, htmlFile, errString)
elif os.path.exists(htmlFileAlter):
return _checkModificationTime(filePath, htmlFileAlter, errString)
else:
return (filePath, 'No preview for this type of file.<br>Expected ' +
htmlFile + " or " + htmlFileAlter, errString, QUrl())
def _runHtmlBuilder(self):
# Build the commond line for Sphinx.
if core.config()['Sphinx']['AdvancedMode']:
htmlBuilderCommandLine = core.config()['Sphinx']['Cmdline']
if sys.platform.startswith('linux'):
# If Linux is used, then subprocess cannot take the whole
# commandline as the name of an executable file. Module shlex
# has to be used to parse commandline.
htmlBuilderCommandLine = shlex.split(htmlBuilderCommandLine)
else:
# For available builder options, refer to: http://sphinx-doc.org/builders.html
htmlBuilderCommandLine = [core.config()['Sphinx']['Executable'],
# Place doctrees in the ``_build`` directory; by default, Sphinx
# places this in _build/html/.doctrees.
'-d', os.path.join('_build', 'doctrees'),
# Source directory -- the current directory, since we'll chdir to
# the project directory before executing this.
'.',
# Build directory
core.config()['Sphinx']['OutputPath']]
# Invoke it.
try:
# Clear the log at the beginning of a Sphinx build.
self.logWindowClear.emit()
cwd = core.config()['Sphinx']['ProjectPath']
# If the command line is already a string (advanced mode), just print it.
# Otherwise, it's a list that should be transformed to a string.
if isinstance(htmlBuilderCommandLine, str):
htmlBuilderCommandLineStr = htmlBuilderCommandLine
else:
htmlBuilderCommandLineStr = ' '.join(htmlBuilderCommandLine)
self.logWindowText.emit('{} : {}\n\n'.format(cwd,
htmlBuilderCommandLineStr))
# Run Sphinx, reading stdout in a separate thread.
self._qe = QEventLoop()
# Sphinx will output just a carriage return (0x0D) to simulate a
# single line being updated by build status and the build
# progresses. Without universal newline support here, we'll wait
# until the build is complete (with a \n\r) to report any build
# progress! So, enable universal newlines, so that each \r will be
# treated as a separate line, providing immediate feedback on build
# progress.
popen = open_console_output(htmlBuilderCommandLine, cwd=cwd,
universal_newlines=True)
# Perform reads in an event loop. The loop is exit when all reads
# have completed. We can't simply start the _stderr_read thread
# here, because calls to self._qe_exit() will be ignored until
# we're inside the event loop.
QTimer.singleShot(0, lambda: self._popen_read(popen))
self._qe.exec_()
except OSError as ex:
return (
'Failed to execute HTML builder:\n'
'{}\n'.format(str(ex)) +
'Go to Settings -> Settings -> CodeChat to set HTML'
' builder configurations.')
return self._stderr
# Read from stdout (in this thread) and stderr (in another thread),
# so that the user sees output as the build progresses, rather than only
# producing output after the build is complete.
def _popen_read(self, popen):
# Read are blocking; we can't read from both stdout and stderr in the
# same thread without possible buffer overflows. So, use this thread to
# read from and immediately report progress from stdout. In another
# thread, read all stderr and report that after the build finishes.
self._ac.start(None, self._stderr_read, popen.stderr)
# Read a line of stdout then report it to the user immediately.
s = popen.stdout.readline()
while s:
self.logWindowText.emit(s.rstrip('\n'))
s = popen.stdout.readline()
self._SphinxInvocationCount += 1
# I would expect the following code to do the same thing. It doesn't:
# instead, it waits until Sphinx completes before returning anything.
# ???
#
# .. code-block: python
# :linenos:
#
# for s in popen.stdout:
# self.logWindowText.emit(s)
# Runs in a separate thread to read stdout. It then exits the QEventLoop as
# a way to signal that stderr reads have completed.
def _stderr_read(self, stderr):
self._stderr = stderr.read()
self._qe.exit()
class Kiwoom(QAxWidget):
def __init__(self):
super().__init__()
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
# variables
self.login_event_loop = None
self.tr_event_loop = None
self.prev_next = None
self.ohlcv = {
'date': [],
'open': [],
'high': [],
'low': [],
'close': [],
'volume': []
}
self.opw00001_data = ""
self.opw00018_data = {'single': [], 'multi': []}
# signal & slot
self.OnEventConnect.connect(self.onEventConnect)
self.OnReceiveTrData.connect(self.onReceiveTrData)
self.OnReceiveChejanData.connect(self.onReceiveTradeData)
""" ======= METHODS ======= """
def changeFormat(self, strData, percent=0):
"""
Change string to currency format
:param strData: 바꿀 문자열
:param percent: 0=정수, 1=소수점, 2=백분율
"""
is_minus = False
if strData.startswith('-'):
is_minus = True
strip_str = strData.lstrip('-0')
if strip_str == '':
if percent == 1:
return '0.00'
else:
return '0'
if percent == 1:
ret = format(int(strip_str) / 100, ',.2f')
elif percent == 2:
ret = format(float(strip_str), ',.2f')
else:
ret = format(int(strip_str), ',d')
if ret.startswith('.'):
ret = '0' + ret
if is_minus:
ret = '-' + ret
return ret
""" ======= OpenAPI+ METHODS ======= """
def commConnect(self):
""" execute login window """
self.dynamicCall("CommConnect()")
self.login_event_loop = QEventLoop()
self.login_event_loop.exec_()
def commRqData(self, sRQName, sTrCode, nPrevNext, sScreenNo):
"""
transmit Transaction to server
:param nPrevNext: 0=조회, 2=연속
"""
self.dynamicCall("CommRqData(QString, QString, int, QString)", sRQName,
sTrCode, nPrevNext, sScreenNo)
self.tr_event_loop = QEventLoop()
self.tr_event_loop.exec_()
def getLoginInfo(self, sTag):
ret = self.dynamicCall("GetLoginInfo(QString)", sTag)
return ret
def sendOrder(self, sRQName, sScreenNo, sAccountNo, nOrderType, sItemCode,
nQty, nPrice, sBid, sOrgOrderNo):
ret = self.dynamicCall(
"SendOrder(QString, QString, QString, int, QString, int, int, QString, QString)",
[
sRQName, sScreenNo, sAccountNo, nOrderType, sItemCode, nQty,
nPrice, sBid, sOrgOrderNo
])
print("SEND ORDER ERROR : " + str(ret))
return ret
def setInputValue(self, sID, sValue):
self.dynamicCall("SetInputValue(QString, QString)", sID, sValue)
def commGetData(self, sItemCode, sRealType, sFieldName, nIndex,
sInnerFieldName):
"""
- Transaction 데이터
:param sItemCode : Tr명
:param sRealType : not used
:param sFieldName : 레코드 명
:param nIndex : 반복 인뎃스
:param sInnerFieldName : 아이템 명
- 실시간 데이터
:param sItemCode : Key Code
:param sRealType : Real Type
:param sFieldName : Item Index
:param nIndex : not used
:param sInnerFieldName : not used
- 체결잔고 데이터
:param sItemCode : 체결구분
:param sRealType : "-1"
:param sFieldName : not used
:param nIndex : Item Index
:param sInnerFieldName : not used
"""
ret = self.dynamicCall(
"CommGetData(QString, QString, QString, int, QString)", sItemCode,
sRealType, sFieldName, nIndex, sInnerFieldName)
return ret.strip()
def disconnectRealData(self, sScreenNo):
pass
def getRepeatCnt(self, sTrCode, sRecordName):
""" 레코드 반복 횟수 리턴"""
ret = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode,
sRecordName)
return ret
def getCodeListByMarket(self, sMarket):
""" 시장구분에 따른 종목코드 리턴"""
ret = self.dynamicCall("GetCodeListByMarket(QString)", sMarket)
item_code = ret.split(';')
return item_code
def getConnectState(self):
ret = self.dynamicCall("GetConnectState()")
return ret
def getMasterCodeName(self, sItemCode):
""" 종목코드 한글명 리턴 """
ret = self.dynamicCall("GetMasterCodeName(QString)", sItemCode)
return ret
def getMasterListedStockCnt(self, sItemCode):
""" 종목코드의 상장주식 수를 리턴"""
pass
def getCommData(self, sTrCode, sRecordName, nIndex, sItemName):
"""
cf) commGetData
개발가이드에서 CommGetData보다 제약이 적은 GetCommData를 쓰라고 명시
ex)현재가출력 : openAPI.GetCommData("opt00001_req", "주식기본정보", 0, "현재가")
"""
ret = self.dynamicCall("GetCommData(QString, QString, int, QString)",
sTrCode, sRecordName, nIndex, sItemName)
return ret
def getTradeData(self, nFid):
""" 체결잔고 데이터 리턴"""
ret = self.dynamicCall("GetChejanData(int)", nFid)
return ret
"""
======= OpenAPI+ EVENT FUNCTIONS
시세조회나 조건검색요청등 서버에 데이터를 요청해서 키움OpenAPI가 데이터를 수신하게 되면 사용자 프로그램에 특정함수를 자동으로 호출해서 알려줍니다.
이렇게 어떤 조건이 맞으면 키움OpenAPI에서 자동으로 호출해서 알려주며 OnReceiveTRData(), OnEventConnect(), OnReceiveRealData()
처럼 함수이름 앞에 On~이 붙어있어서 일반적인 함수와 구별됩니다.
로그인, 시세조회, 주문등 각 요청별로 요청함수, 이벤트와 데이터획득함수가 분류되어 있으니 용도에 맞는 함수를 사용해야 합니다.
"""
def onReceiveTrData(self, sScreenNo, sRQname, sTrCode, sRecordName,
sPrevNext, notUsed1, notUsed2, notUsed3, notUsed4):
"""
서버로부터 event를 받으면 자동 실행된다.
수신 받은 데이터는 commGetData()를 통해 받는다.
GetCommData()를 쓰라고 하는데 자세한건 나중에 수정
sPrevNext: 연속 조회가 필요하면 2를 리턴 받음
"""
self.prev_next = sPrevNext
print("RqName : " + sRQname)
print("Trcode : " + sTrCode + "\n")
if sRQname == "opt10081_req": # opt10081 : 주식일봉차트조회요청
cnt = self.getRepeatCnt(sTrCode, sRQname)
for i in range(cnt):
date = self.commGetData(sTrCode, "", sRQname, i, "일자")
open = self.commGetData(sTrCode, "", sRQname, i, "시가")
high = self.commGetData(sTrCode, "", sRQname, i, "고가")
low = self.commGetData(sTrCode, "", sRQname, i, "저가")
close = self.commGetData(sTrCode, "", sRQname, i, "현재가")
self.ohlcv['date'].append(date)
self.ohlcv['open'].append(int(open))
self.ohlcv['high'].append(int(high))
self.ohlcv['low'].append(int(low))
self.ohlcv['close'].append(int(close))
elif sRQname == "opw00001_req": # opw00001 : 예수금상세현황요청
estimated_day2_deposit = self.commGetData(sTrCode, "", sRQname, 0,
"d+2추정예수금")
self.opw00001_data = self.changeFormat(estimated_day2_deposit)
elif sRQname == "opw00018_req": # opw00018 : 계좌평가잔고내역
"""
single : 개별
multi : 개별합산
한 번의 TR요청으로 최대 20개 보유 종목 가져옴
"""
# single data
single_data = []
total_purchase_price = self.commGetData(sTrCode, "", sRQname, 0,
"총매입금액")
single_data.append(self.changeFormat(total_purchase_price))
total_eval_price = self.commGetData(sTrCode, "", sRQname, 0,
"총평가금액")
single_data.append(self.changeFormat(total_eval_price))
total_eval_profit_loss_price = self.commGetData(
sTrCode, "", sRQname, 0, "총평가손익금액")
single_data.append(self.changeFormat(total_eval_profit_loss_price))
total_profit_rate = self.commGetData(sTrCode, "", sRQname, 0,
"총수익률(%)")
single_data.append(self.changeFormat(total_profit_rate, 1))
estimated_deposit = self.commGetData(sTrCode, "", sRQname, 0,
"추정예탁자산")
single_data.append(self.changeFormat(estimated_deposit))
self.opw00018_data['single'] = single_data
# multi data
cnt = self.getRepeatCnt(sTrCode, sRQname)
for i in range(cnt):
multi_data = []
item_name = self.commGetData(sTrCode, "", sRQname, i, "종목명")
multi_data.append(item_name)
quantity = self.commGetData(sTrCode, "", sRQname, i, "보유수량")
multi_data.append(self.changeFormat(quantity))
purchase_price = self.commGetData(sTrCode, "", sRQname, i,
"매입가")
multi_data.append(self.changeFormat(purchase_price))
current_price = self.commGetData(sTrCode, "", sRQname, i,
"현재가")
multi_data.append(self.changeFormat(current_price))
eval_profit_loss_price = self.commGetData(
sTrCode, "", sRQname, i, "평가손익")
multi_data.append(self.changeFormat(eval_profit_loss_price))
profit_rate = self.commGetData(sTrCode, "", sRQname, i,
"수익률(%)")
multi_data.append(self.changeFormat(profit_rate, 2))
self.opw00018_data['multi'].append(multi_data)
self.tr_event_loop.exit()
def onReceiveRealData(self, sItemCode):
pass
def onReceiveMsg(self, sScrNo, sRQname, sTrCode, sMsg):
pass
def onReceiveTradeData(self, sTradeType, nItemCnt, sFidList):
print("sTradeType : ", sTradeType)
print(self.getTradeData(9203))
print(self.getTradeData(302))
print(self.getTradeData(900))
print(self.getTradeData(909))
def onEventConnect(self, nErrCode):
if nErrCode == 0:
print("connected")
else:
print("disconnected")
self.login_event_loop.exit()
def onReceiveCondition(self, sItemCode, sType, sConditionName,
sConditionIndex):
"""
조건검색 실시간 편입, 이탈 종목을 받을 시점을 알려준다
:param sItemCode:
:param sType: 'I' = 편입, 'D' = 이탈
:return:
"""
pass
class KFOpenAPI(QAxWidget):
def __init__(self):
super().__init__()
self.setControl("KFOPENAPI.KFOpenAPICtrl.1")
self.loginLoop = None
self.requestLoop = None
self.orderLoop = None
self.error = None
self.orderNo = ""
self.inquiry = 0
self.msg = ""
self.screenNo = 1000
self.dicScrNo = {}
self.OnReceiveTrData.connect(self.ReceiveTrData)
self.OnReceiveRealData.connect(self.ReceiveRealData)
self.OnReceiveMsg.connect(self.ReceiveMsg)
self.OnReceiveChejanData.connect(self.ReceiveChejanData)
self.OnEventConnect.connect(self.EventConnect)
print("Initialized Kiwoom OpenAPI-W")
###############################################################
# Method
###############################################################
def CommConnect(self):
"""
1) CommConnect
원형 LONG CommConnect(LONG nAutoUpgrade)
설명 로그인 윈도우를 실행한다.
입력값 0 – 버전 수동처리, 1 – 버전 자동처리
*로그인창 및 OCX 파일을 버전처리 받는 경우에,
수동처리시, 고객 프로그램(ocx포함)을 직접 수동으로 Close하고 버전처리 진행
자동처리시, 고객 프로그램(ocx포함)을 자동으로 Close하고 버전처리 및 자동
재실행을 함.
반환값 0 - 성공, 음수값은 실패
비고 로그인이 성공하거나 실패하는 경우 OnEventConnect 이벤트가 발생하고
이벤트의 인자 값으로 로그인 성공 여부를 알 수 있다.
구분자 자동여부는 고객 프로그램(ocx포함)이 단독으로 사용되어지는 경우에,
구분자를 자동으로 선택하시고, 고객 프로그램(ocx포함)이 다른프로그램과
연동되어져 실행시 데이터를 수신해야하는 경우에는 구분자를 수동으로 선택.
"""
self.dynamicCall("CommConnect(1)")
self.loginEventLoop = QEventLoop()
self.loginEventLoop.exec_()
def CommRqData(self, sRQName, sTrCode, sPrevNext, sScreenNo):
"""
2) CommRqData
원형 LONG CommRqData (BSTR sRQName, BSTR sTrCode, BSTR sPrevNext, BSTR sScreenNo )
설명 Tran을 서버로 송신한다.
입력값 BSTR sRQName BSTR sTrCode long nPrevNext BSTR sScreenNo
반환값 OP_ERR_SISE_OVERFLOW – 과도한 시세조회로 인한 통신불가
OP_ERR_RQ_STRUCT_FAIL – 입력 구조체 생성 실패
OP_ERR_RQ_STRING_FAIL – 요청전문 작성 실패
OP_ERR_NONE – 정상처리
비고 sRQName – 사용자구분 명
sTrCode - Tran명 입력
nPrevNext – 서버에서 내려준 Next키값 입력(샘플참조)
sScreenNo - 4자리의 화면번호(1~9999 :숫자값으로만 가능)
Ex) openApi.CommRqData( “RQ_1”, “OPT00001”, “”, “0101”);
"""
if not self.GetConnectState():
print("Error : KiwoomConnectError by CommRqData")
raise KiwoomConnectError()
if not (isinstance(sRQName, str) and isinstance(sTrCode, str)
and isinstance(sPrevNext, str) and isinstance(sScreenNo, str)):
print("Error : ParameterTypeError by CommRqData")
raise ParameterTypeError()
errorCode = self.dynamicCall(
'CommRqData(QString, QString, QString, QString)', sRQName, sTrCode,
sPrevNext, sScreenNo)
if errorCode != ErrorCode.OP_ERR_NONE:
raise KiwoomProcessingError("CommRqData(): " +
ErrorCode.CAUSE[errorCode])
else:
self.requestLoop = QEventLoop()
self.requestLoop.exec_()
def SetInputValue(self, sID, sValue):
"""
3) SetInputValue
원형 void SetInputValue(BSTR sID, BSTR sValue)
설명 Tran 입력 값을 서버통신 전에 입력한다.
입력값 sID – 아이템명 sValue – 입력 값
반환값 없음
비고 통신 Tran 매뉴얼 참고
Ex) openApi.SetInputValue(“종목코드”, “6AH16”);
openApi.SetInputValue(“계좌번호”, “5015123401”);
"""
if not (isinstance(sID, str) and isinstance(sValue, str)):
print("Error : ParameterTypeError by SetInputValue")
raise ParameterTypeError()
self.dynamicCall("SetInputValue(QString, QString)", sID, sValue)
def GetCommData(self, strTrCode, strRecordName, nIndex, strItemName):
"""
4) GetCommData
원형 BSTR GetCommData(BSTR strTrCode, BSTR strRecordName, long nIndex, BSTR strItemName)
설명 수신 데이터를 반환한다.
입력값 strTrCode – Tran 코드
strRecordName – 레코드명
nIndex – 복수데이터 인덱스
strItemName – 아이템명
반환값 수신 데이터
비고 Ex)현재가출력 - openApi.GetCommData(“OPT00001”, “해외선물기본정보”, 0, “현재가”);
"""
if not (isinstance(strTrCode, str) and isinstance(strRecordName, str)
and isinstance(nIndex, int) and isinstance(strItemName, str)):
print("Error : ParameterTypeError by GetCommData")
raise ParameterTypeError()
return self.dynamicCall("GetCommData(QString, QString, int, QString)",
strTrCode, strRecordName, nIndex,
strItemName).strip()
def CommTerminate(self):
"""
5) CommTerminate
원형 void CommTerminate()
설명 OpenAPI의 서버 접속을 해제한다.
입력값 없음
반환값 없음
비고 통신 연결 상태는 GetConnectState 메소드로 알 수 있다.
"""
self.dynamicCall("CommTerminate()")
def GetRepeatCnt(self, sTrCode, sRecordName):
"""
6) GetRepeatCnt
원형 LONG GetRepeatCnt(BSTR sTrCode, BSTR sRecordName)
설명 레코드 반복횟수를 반환한다.
입력값 sTrCode – Tran 명
sRecordName – 레코드 명
반환값 레코드의 반복횟수
비고 Ex) openApi.GetRepeatCnt(“OPT00001”, “해외선물체결데이타”);
"""
if not (isinstance(sTrCode, str) and isinstance(sRecordName, str)):
print("Error : ParameterTypeError by GetRepeatCnt")
raise ParameterTypeError()
return self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode,
sRecordName)
def DisconnectRealData(self, sScrNo):
"""
7) DisconnectRealData
원형 void DisconnectRealData(BSTR sScnNo)
설명 화면 내 모든 리얼데이터 요청을 제거한다.
입력값 sScrNo – 화면번호[4]
반환값 없음
비고 화면을 종료할 때 반드시 위 함수를 호출해야 한다.
Ex) openApi.DisconnectRealData(“0101”);
"""
if not (isinstance(sScrNo, str)):
print("Error : ParameterTypeError by DisconnectRealData")
raise ParameterTypeError()
self.dynamicCall("DisconnectRealData(QString)", sScrNo)
def GetCommRealData(self, strRealType, nFid):
"""
8) GetCommRealData
원형 BSTR GetCommRealData(BSTR strRealType, long nFid)
설명 실시간데이터를 반환한다.
입력값 strRealType – 실시간 구분
nFid – 실시간 아이템
반환값 수신 데이터
비고 Ex) 현재가출력 - openApi.GetCommRealData(“해외선물시세”, 10);
"""
if not (isinstance(strRealType, str) and isinstance(nFid, int)):
print("Error : ParameterTypeError by GetCommRealData")
raise ParameterTypeError()
return self.dynamicCall("GetCommRealData(QString, int)", strRealType,
nFid)
def GetChejanData(self, nFid):
"""
9) GetChejanData
원형 BSTR GetChjanData(long nFid)
설명 체결잔고 데이터를 반환한다.
입력값 nFid – 체결잔고 아이템
반환값 수신 데이터
비고 Ex) 현재가출력 – openApi.GetChejanData(910); //체결가격
"""
if not (isinstance(nFid, int)):
print("Error : ParameterTypeError by GetChejanData")
raise ParameterTypeError()
return self.dynamicCall("GetChejanData(int)", nFid)
def SendOrder(self, sRQName, sScreenNo, sAccNo, nOrderType, sCode, nQty,
sPrice, sStop, sHogaGb, sOrgOrderNo):
"""
10) SendOrder
원형 LONG SendOrder(BSTR sRQName, BSTR sScreenNo, BSTR sAccNo, LONG nOrderType, BSTR sCode, LONG nQty,
BSTR sPrice, BSTR sStop, BSTR sHogaGb, BSTR sOrgOrderNo )
설명 주문을 서버로 전송한다.
입력값 sRQName - 사용자 구분 요청 명
sScreenNo - 화면번호[4] (1~9999 :숫자값으로만 가능)
sAccNo - 계좌번호[10]
nOrderType - 주문유형 (1:신규매도, 2:신규매수, 3:매도취소, 4:매수취소, 5:매도정정, 6:매수정정)
sCode - 종목코드
nQty – 주문수량
sPrice – 주문단가
sStop - Stop단가
sHogaGb - 거래구분
sOrgOrderNo – 원주문번호
반환값 에러코드 <7.에러코드표 참고>
비고 sHogaGb – 1:시장가, 2:지정가, 3:STOP, 4:STOP LIMIT
ex) 지정가 매수 - openApi.SendOrder(“RQ_1”, “0101”, “5015123410”, 2, “6AH16”, 10, “0.7900”, “2”, “”);
시장가 매수 - openApi.SendOrder(“RQ_1”, “0101”, “5015123410”, 2, “6AH16”, 10, “0”, “1”, “”);
매수 정정 - openApi.SendOrder(“RQ_1”,“0101”, “5015123410”, 6, “6AH16”, 10, “0.7800”, “0”, “200060”);
매수 취소 - openApi.SendOrder(“RQ_1”, “0101”, “5015123410”, 4, “6AH16”, 10, “0”, “0”, “200061”);
"""
if not self.getConnectState():
print("Error : KiwoomConnectError by SendOrder")
raise KiwoomConnectError()
if not (isinstance(sRQName, str) and isinstance(sScreenNo, str)
and isinstance(sAccNo, str) and isinstance(nOrderType, int)
and isinstance(sCode, str) and isinstance(nQty, int)
and isinstance(sPrice, str) and isinstance(sStop, str)
and isinstance(sHogaGb, str) and isinstance(sOrgOrderNo, str)):
print("Error : ParameterTypeError by SendOrder")
raise ParameterTypeError()
ErrorCode = self.dynamicCall(
"SendOrder(QString, QString, QString, int, QString, int, QString, QString, QString, QString)",
sRQName, sScreenNo, sAccNo, nOrderType, sCode, nQty, sPrice, sStop,
sHogaGb, sOrgOrderNo)
if ErrorCode != ErrorCode.OP_ERR_NONE:
raise KiwoomProcessingError("sendOrder(): " +
ErrorCode.CAUSE[ErrorCode])
self.orderLoop = QEventLoop()
self.orderLoop.exec_()
def GetLoginInfo(self, sTag):
"""
11) GetLoginInfo
원형 BSTR GetLoginInfo(BSTR sTag)
설명 로그인한 사용자 정보를 반환한다.
입력값 BSTR sTag : 사용자 정보 구분 TAG값 (비고)
반환값 TAG값에 따른 데이터 반환
비고 BSTR sTag에 들어 갈 수 있는 값은 아래와 같음
“ACCOUNT_CNT” – 전체 계좌 개수를 반환한다.
"ACCNO" – 전체 계좌를 반환한다. 계좌별 구분은 ‘;’이다.
“USER_ID” - 사용자 ID를 반환한다.
“USER_NAME” – 사용자명을 반환한다.
“KEY_BSECGB” – 키보드보안 해지여부. 0:정상, 1:해지
“FIREW_SECGB” – 방화벽 설정 여부. 0:미설정, 1:설정, 2:해지
Ex) openApi.GetLoginInfo(“ACCOUNT_CNT”);
"""
if not self.GetConnectState():
print("Error : KiwoomConnectError by GetLoginInfo")
raise KiwoomConnectError()
if not isinstance(sTag, str):
print("Error : ParameterTypeError by GetLoginInfo")
raise ParameterTypeError()
if sTag not in [
"ACCOUNT_CNT", "ACCNO", "USER_ID", "USER_NAME",
"GetServerGubun"
]:
print("Error : ParameterValueError by GetLoginInfo")
raise ParameterValueError()
return self.dynamicCall("GetLoginInfo(QString)", sTag)
def GetGlobalFutureItemlist(self):
"""
12) GetGlobalFutureItemlist
원형 BSTR GetGlobalFutureItemlist()
설명 해외선물 상품리스트를 반환한다.
입력값 없음
반환값 해외선물 상품리스트, 상품간 구분은 ‘;’이다.
비고 해외선물 상품리스트(6A, 6B, 6C, ES…..)
"""
return self.dynamicCall("GetGlobalFutureItemlist()")
def GetGlobalOptionItemlist(self):
"""
13) GetGlobalOptionItemlist
원형 BSTR GetGlobalOptionItemlist()
설명 해외옵션 상품리스트를 반환한다.
입력값 없음
반환값 해외옵션 상품리스트, 상품간 구분은 ‘;’이다.
비고 해외옵션 상품리스트(6A, 6B, 6C, ES…..)
"""
return self.dynamicCall("GetGlobalOptionItemlist()")
def GetGlobalFutureCodelist(self, sItem):
"""
14) GetGlobalFutureCodelist
원형 BSTR GetGlobalFutureCodelist(BSTR sItem)
설명 해외상품별 해외선물 종목코드리스트를 반환한다.
입력값 해외상품
반환값 해외선물 종목코드리스트, 종목간 구분은 ‘;’이다.
비고 해외선물 종목코드리스트
"""
if not (isinstance(sItem, str)):
print("Error : ParameterTypeError by GetGlobalFutureCodelist")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalFutureCodelist(QString)", sItem)
def GetGlobalOptionCodelist(self, sItem):
"""
15) GetGlobalOptionCodelist
원형 BSTR GetGlobalOptionCodelist(BSTR sItem)
설명 해외상품별 해외선물 종목코드리스트를 반환한다.
입력값 해외상품
반환값 해외옵션 종목코드리스트, 종목간 구분은 ‘;’이다.
비고 해외옵션 종목코드리스트
"""
if not (isinstance(sItem, str)):
print("Error : ParameterTypeError by GetGlobalOptionCodelist")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalOptionCodelist(QString)", sItem)
def GetConnectState(self):
"""
16) GetConnectState
원형 LONG GetConnectState()
설명 현재접속상태를 반환한다.
입력값 없음
반환값 접속상태
비고 0:미연결, 1:연결완료
"""
return self.dynamicCall("GetConnectState()")
def GetAPIModulePath(self):
"""
17) GetAPIModulePath
원형 BSTR GetAPIModulePath()
설명 OpenAPI모듈의 경로를 반환한다.
입력값 없음
반환값 경로
비고
"""
return self.dynamicCall("GetAPIModulePath()")
def GetCommonFunc(self, sFuncName, sParam):
"""
18) GetCommonFunc
원형 BSTR GetCommonFunc(BSTR sFuncName, BSTR sParam)
설명 공통함수로 추후 추가함수가 필요시 사용할 함수이다.
입력값 함수명, 인자값
반환값 문자값으로 반환한다.
비고
"""
if not (isinstance(sFuncName, str) and isinstance(sParam, str)):
print("Error : ParameterTypeError by GetCommonFunc")
raise ParameterTypeError()
return self.dynamicCall("GetCommonFunc(QString, QString)", sFuncName,
sParam)
def GetConvertPrice(self, sCode, sPrice, nType):
"""
19) GetConvertPrice
원형 BSTR GetConvertPrice(BSTR sCode, BSTR sPrice, LONG nType)
설명 가격 진법에 따라 변환된 가격을 반환한다.
입력값 종목코드, 가격, 타입(0 : 진법->10진수, 1 : 10진수->진법)
반환값 문자값으로 반환한다.
비고
"""
if not (isinstance(sCode, str) and isinstance(sPrice, str)
and isinstance(nType, str)):
print("Error : ParameterTypeError by GetConvertPrice")
raise ParameterTypeError()
return self.dynamicCall("GetConvertPrice(QString, QString, int)",
sCode, sPrice, nType)
def GetGlobalFutOpCodeInfoByType(self, nGubun, sType):
"""
20) GetGlobalFutOpCodeInfoByType
원형 BSTR GetGlobalFutCodeInfoByType(LONG nGubun, BSTR sType)
설명 해외선물옵션종목코드정보를 타입별로 반환한다.
입력값 nGubun : 0(해외선물), 1(해외옵션)
sType : IDX(지수), CUR(통화), INT(금리), MLT(금속), ENG(에너지), CMD(농산물)
반환값 종목코드정보리스트들을 문자값으로 반환한다.(아래 종목마스터파일 참조)
비고 전체는 “”으로 보내면 된다.
"""
if not (isinstance(nGubun, int) and isinstance(sType, str)):
print("Error : ParameterTypeError by GetGlobalFutOpCodeInfoByType")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalFutOpCodeInfoByType(int, QString)",
nGubun, sType)
def GetGlobalFutOpCodeInfoByCode(self, sCode):
"""
21) GetGlobalFutOpCodeInfoByCode
원형 BSTR GetGlobalFutCodeInfoByCode(BSTR sCode)
설명 해외선물옵션종목코드정보를 종목코드별로 반환한다.
입력값 sCode : 해외선물옵션 종목코드 입력
반환값 종목코드정보를 문자값으로 반환한다.(아래 종목마스터파일 참조)
비고
"""
if not (isinstance(sCode, str)):
print("Error : ParameterTypeError by GetGlobalFutOpCodeInfoByCode")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalFutOpCodeInfoByCode(QString)", sCode)
def GetGlobalFutureItemlistByType(self, sType):
"""
22) GetGlobalFutureItemlistByType
원형 BSTR GetGlobalFutureItemlistByType (BSTR sType)
설명 해외선물상품리스트를 타입별로 반환한다.
입력값 sType : IDX(지수), CUR(통화), INT(금리), MLT(금속), ENG(에너지), CMD(농산물)
반환값 상품리스트를 문자값으로 반환한다.
비고
"""
if not (isinstance(sType, str)):
print(
"Error : ParameterTypeError by GetGlobalFutureItemlistByType")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalFutureItemlistByType(QString)",
sType)
def GetGlobalFutureCodeByItemMonth(self, sItem, sMonth):
"""
23) GetGlobalFutureCodeByItemMonth
원형 BSTR GetGlobalFutureCodeByItemMonth(BSTR sItem, BSTR sMonth)
설명 해외선물종목코드를 상품/월물별로 반환한다.
입력값 sItem: 상품코드(6A, ES..),
sMonth: “201606”
반환값 종목코드를 문자값으로 반환한다.
비고
"""
if not (isinstance(sItem, str) and isinstance(sMonth, str)):
print(
"Error : ParameterTypeError by GetGlobalFutureCodeByItemMonth")
raise ParameterTypeError()
return self.dynamicCall(
"GetGlobalFutureCodeByItemMonth(QString, QString)", sItem, sMonth)
def GetGlobalOptionCodeByMonth(self, sItem, sCPGubun, sActPrice, sMonth):
"""
24) GetGlobalOptionCodeByMonth
원형 BSTR GetGlobalOptionCodeByMonth(BSTR sItem, BSTR sCPGubun, BSTR sActPrice, BSTR sMonth)
설명 해외옵션종목코드를 상품/콜풋/행사가/월물별로 반환한다.
입력값 sItem: 상품코드(6A, ES..),
sCPGubun: C(콜)/P(풋), sActPrice: 0.760,
sMonth: “201606”
반환값 종목코드를 문자값으로 반환한다.
비고
"""
if not (isinstance(sItem, str) and isinstance(sCPGubun, str)
and isinstance(sActPrice, str) and isinstance(sMonth, str)):
print("Error : ParameterTypeError by GetGlobalOptionCodeByMonth")
raise ParameterTypeError()
return self.dynamicCall(
"GetGlobalOptionCodeByMonth(QString, QString, QString, QString)",
sItem, sCPGubun, sActPrice, sMonth)
def GetGlobalOptionMonthByItem(self, sItem):
"""
25) GetGlobalOptionMonthByItem
원형 BSTR GetGlobalOptionMonthByItem(BSTR sItem)
설명 해외옵션월물리스트를 상품별로 반환한다.
입력값 sItem: 상품코드(6A, ES..)
반환값 월물리스트를 문자값으로 반환한다.
비고
"""
if not (isinstance(sItem, str)):
print("Error : ParameterTypeError by GetGlobalOptionMonthByItem")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalOptionMonthByItem(QString", sItem)
def GetGlobalOptionActPriceByItem(self, sItem):
"""
26) GetGlobalOptionActPriceByItem
원형 BSTR GetGlobalOptionActPriceByItem(BSTR sItem)
설명 해외옵션행사가리스트를 상품별로 반환한다.
입력값 sItem: 상품코드(6A, ES..)
반환값 행사가리스트를 문자값으로 반환한다.
비고
"""
if not (isinstance(sItem, str)):
print(
"Error : ParameterTypeError by GetGlobalOptionActPriceByItem")
raise ParameterTypeError()
return self.dynamicCall("GetGlobalOptionActPriceByItem(QString", sItem)
def GetGlobalFutureItemTypelist(self):
"""
27) GetGlobalFutureItemTypelist
원형 BSTR GetGlobalFutureItemTypelist()
설명 해외선물상품타입리스트를 반환한다.
입력값 없음
반환값 상품타입리스트를 문자값으로 반환한다.
비고 IDX;CUR;INT;MLT;ENG;CMD; 반환
"""
return self.dynamicCall("GetGlobalFutureItemTypelist()")
def GetCommFullData(self, strTrCode, strRecordName, nGubun):
"""
28) GetCommFullData
원형 BSTR GetCommFullData(BSTR strTrCode, BSTR strRecordName, LONG nGubun )
설명 수신된 전체데이터를 반환한다.
입력값 strTrCode – Tran 코드
strRecordName – 레코드명
nGubun– 0 : 전체(싱글+멀티), 1 : 싱글데이타, 2 : 멀티데이타
반환값 수신 전체데이터를 문자값으로 반환한다.
비고 WKOAStudio의 TR목록탭에서 필드 사이즈 참조.(필드명 옆 가로안의 값들)
모든 시세/원장 조회에 사용 가능하며, 특히 차트데이타 같은 대용량 데이터를
한번에 받아서 처리가능.
"""
if not (isinstance(strTrCode, str) and isinstance(strRecordName, str)):
print("Error : ParameterTypeError by GetCommFullData")
raise ParameterTypeError()
return self.dynamicCall("GetCommFullData(QString, QString", strTrCode,
strRecordName)
def GetScreenNumber(self):
ret = "{0:04d}".format(self.screenNo)
if self.screenNo == 9999:
raise MaxScreenNumber()
self.dicScrNo.update({ret: self.screenNo})
self.screenNo += 1
return ret
###############################################################
# Event
# 각종 수신 Event를 처리
###############################################################
# OnReceiveTrData
def ReceiveTrData(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
"""
1) OnReceiveTrData
원형 void OnReceiveTrData(BSTR sScrNo, BSTR sRQName, BSTR sTrCode, BSTR sRecordName, BSTR sPreNext)
설명 서버통신 후 데이터를 받은 시점을 알려준다.
입력값 sScrNo – 화면번호
sRQName – 사용자구분 명
sTrCode – Tran 명
sRecordName – Record 명
sPreNext – 연속조회 유무
반환값 없음
비고 sRQName – CommRqData의 sRQName과 매핑되는 이름이다.
sTrCode – CommRqData의 sTrCode과 매핑되는 이름이다.
"""
print("OnReceiveTrData")
self.requestLoop.exit()
# OnReceiveRealData
def ReceiveRealData(self, sJongmokCode, sRealType, sRealData):
"""
2) OnReceiveRealData
원형 void OnReceiveRealData(BSTR sJongmokCode, BSTR sRealType, BSTR sRealData)
설명 실시간데이터를 받은 시점을 알려준다.
입력값 sJongmokCode – 종목코드
sRealType – 리얼타입
sRealData – 실시간 데이터전문
반환값 없음
비고
"""
#print("OnReceiveRealData")
# OnReceiveMsg
def ReceiveMsg(self, sScrNo, sRQName, sTrCode, sMsg):
"""
3) OnReceiveMsg
원형 void OnReceiveMsg(BSTR sScrNo BSTR sRQName, BSTR sTrCode, BSTR sMsg)
설명 서버통신 후 메시지를 받은 시점을 알려준다.
입력값 sScrNo – 화면번호
sRQName – 사용자구분 명
sTrCode – Tran 명
sMsg – 서버메시지
반환값 없음
비고 sScrNo – CommRqData의 sScrNo와 매핑된다.
sRQName – CommRqData의 sRQName 와 매핑된다.
sTrCode – CommRqData의 sTrCode 와 매핑된다
"""
print("OnReceiveMsg")
self.msg += sRQName + ": " + sMsg + "\r\n\r\n"
# OnReceiveChejanData
def ReceiveChejanData(self, sGubun, nItemCnt, sFidList):
"""
4) OnReceiveChejanData
원형 void OnReceiveChejanData(BSTR sGubun, LONG nItemCnt, BSTR sFidList)
설명 체결데이터를 받은 시점을 알려준다.
입력값 sGubun – 체결구분
nItemCnt - 아이템갯수
sFidList – 데이터리스트
sSplmMsg - 1.0.0.1 버전 이후 사용하지 않음.
반환값 없음
비고 sGubun – 0:주문체결통보, 1:잔고통보, 3:특이신호
sFidList – 데이터 구분은 ‘;’ 이다
"""
print("OnReceiveChejanData")
if self.requestLoop:
self.requestLoop.exit()
if self.orderLoop:
self.orderLoop.exit()
# 로그인 이벤트 응답
def EventConnect(self, nErrCode):
"""
5) OnEventConnect
원형 void OnEventConnect(LONG nErrCode)
설명 서버 접속 관련 이벤트
입력값 LONG nErrCode : 에러 코드
반환값 없음
비고 nErrCode가 0이면 로그인 성공, 음수면 실패
음수인 경우는 에러 코드 참조
"""
print("OnEventConnect")
self.loginEventLoop.exit()
class CharacterDialog(WindowModalDialog):
def __init__(self, parent):
super(CharacterDialog, self).__init__(parent)
self.setWindowTitle(_("KeepKey Seed Recovery"))
self.character_pos = 0
self.word_pos = 0
self.loop = QEventLoop()
self.word_help = QLabel()
self.char_buttons = []
vbox = QVBoxLayout(self)
vbox.addWidget(WWLabel(CHARACTER_RECOVERY))
hbox = QHBoxLayout()
hbox.addWidget(self.word_help)
for i in range(4):
char_button = CharacterButton('*')
char_button.setMaximumWidth(36)
self.char_buttons.append(char_button)
hbox.addWidget(char_button)
self.accept_button = CharacterButton(_("Accept Word"))
self.accept_button.clicked.connect(partial(self.process_key, 32))
self.rejected.connect(partial(self.loop.exit, 1))
hbox.addWidget(self.accept_button)
hbox.addStretch(1)
vbox.addLayout(hbox)
self.finished_button = QPushButton(_("Seed Entered"))
self.cancel_button = QPushButton(_("Cancel"))
self.finished_button.clicked.connect(partial(self.process_key,
Qt.Key_Return))
self.cancel_button.clicked.connect(self.rejected)
buttons = Buttons(self.finished_button, self.cancel_button)
vbox.addSpacing(40)
vbox.addLayout(buttons)
self.refresh()
self.show()
def refresh(self):
self.word_help.setText("Enter seed word %2d:" % (self.word_pos + 1))
self.accept_button.setEnabled(self.character_pos >= 3)
self.finished_button.setEnabled((self.word_pos in (11, 17, 23)
and self.character_pos >= 3))
for n, button in enumerate(self.char_buttons):
button.setEnabled(n == self.character_pos)
if n == self.character_pos:
button.setFocus()
def is_valid_alpha_space(self, key):
# Auto-completion requires at least 3 characters
if key == ord(' ') and self.character_pos >= 3:
return True
# Firmware aborts protocol if the 5th character is non-space
if self.character_pos >= 4:
return False
return (key >= ord('a') and key <= ord('z')
or (key >= ord('A') and key <= ord('Z')))
def process_key(self, key):
self.data = None
if key == Qt.Key_Return and self.finished_button.isEnabled():
self.data = {'done': True}
elif key == Qt.Key_Backspace and (self.word_pos or self.character_pos):
self.data = {'delete': True}
elif self.is_valid_alpha_space(key):
self.data = {'character': chr(key).lower()}
if self.data:
self.loop.exit(0)
def keyPressEvent(self, event):
self.process_key(event.key())
if not self.data:
QDialog.keyPressEvent(self, event)
def get_char(self, word_pos, character_pos):
self.word_pos = word_pos
self.character_pos = character_pos
self.refresh()
if self.loop.exec_():
self.data = None # User cancelled
class WaitForSignal(unittest.TestCase):
def __init__(self,
# The signal to wait for.
signal,
# The maximum time to wait for the signal to be emitted, in ms.
timeoutMs,
# True to self.assertif the signal wasn't emitted.
assertIfNotRaised=True,
# Expected parameters which this signal must supply.
expectedSignalParams=None,
# True to print exceptions raised in the event loop
printExcTraceback=True,
# Number of times this signal must be emitted
numEmittedExpected=1):
self.signal = signal
self.timeoutMs = timeoutMs
self.expectedSignalParams = expectedSignalParams
self.assertIfNotRaised = assertIfNotRaised
self.printExcTraceback = printExcTraceback
self.numEmittedExpected = numEmittedExpected
# Stores the result of comparing self.expectedSignalParams with the
# actual params.
self.areSenderSignalArgsWrong = False
# The number of times this signal was emitted.
self.numEmitted = 0
# Create a slot which receives a senderSignal with any number
# of arguments. Check the arguments against their expected
# values, if requested, storing the result in senderSignalArgsWrong[0].
# (I can't use senderSignalArgsWrong = True/False, since
# non-local variables cannot be assigned in another scope).
def signalSlot(self, *args):
# If the senderSignal args should be checked and they
# don't match, then they're wrong. In all other cases,
# they're right.
if self.expectedSignalParams:
self.areSenderSignalArgsWrong = (self.expectedSignalParams != args)
self.numEmitted += 1
if self._gotSignal():
# We received the requested signal, so exit the event loop or never
# enter it (exit won't exit an event loop that hasn't been run). When
# this is nested inside other WaitForSignal clauses, signals may be
# received in another QEventLoop, even before this object's QEventLoop
# starts.
self.qe.exit()
# True of the signal was emitted the expected number of times.
def _gotSignal(self):
return self.numEmitted == self.numEmittedExpected
def __enter__(self):
# Create an event loop to run in. Otherwise, we need to use the papp
# (QApplication) main loop, which may already be running and therefore
# unusable.
self.qe = QEventLoop()
# Connect both signals to a slot which quits the event loop.
self.signal.connect(self.signalSlot)
return self
def __exit__(self, exc_type, exc_value, traceback):
# Create a single-shot timer. Could use QTimer.singleShot(),
# but can't cancel this / disconnect it.
self.timer = QTimer()
self.timer.setSingleShot(True)
self.timer.timeout.connect(self.qe.quit)
self.timer.start(self.timeoutMs)
# Catch any exceptions which the EventLoop would otherwise catch
# and not re-raise.
self.exceptions = None
def excepthook(type_, value, tracebackObj):
self.exceptions = (value, tracebackObj)
if self.printExcTraceback:
oldExcHook(type_, value, tracebackObj)
self.qe.exit()
oldExcHook = sys.excepthook
sys.excepthook = excepthook
# Wait for an emitted signal, unless it already occurred.
if not self._gotSignal():
self.qe.exec_()
# Restore the old exception hook
sys.excepthook = oldExcHook
# Clean up: don't allow the timer to call qe.quit after this
# function exits, which would produce "interesting" behavior.
timerIsActive = self.timer.isActive()
self.timer.stop()
# Stopping the timer may not cancel timeout signals in the
# event queue. Disconnect the signal to be sure that loop
# will never receive a timeout after the function exits.
# Likewise, disconnect the senderSignal for the same reason.
self.signal.disconnect(self.signalSlot)
self.timer.timeout.disconnect(self.qe.quit)
# If an exception occurred in the event loop, re-raise it.
if self.exceptions:
value, tracebackObj = self.exceptions
raise value.with_traceback(tracebackObj)
# Check that the signal occurred.
self.sawSignal = timerIsActive and not self.areSenderSignalArgsWrong
if self.assertIfNotRaised:
self.assertTrue(self.sawSignal)
# Don't mask exceptions.
return False
class InstallWizard(QDialog, MessageBoxMixin, BaseWizard):
accept_signal = pyqtSignal()
def __init__(self, config, app, plugins, storage):
BaseWizard.__init__(self, config, plugins, storage)
QDialog.__init__(self, None)
self.setWindowTitle('Electrum - ' + _('Install Wizard'))
self.app = app
self.config = config
# Set for base base class
self.language_for_seed = config.get('language')
self.setMinimumSize(600, 400)
self.accept_signal.connect(self.accept)
self.title = QLabel()
self.main_widget = QWidget()
self.back_button = QPushButton(_("Back"), self)
self.back_button.setText(
_('Back') if self.can_go_back() else _('Cancel'))
self.next_button = QPushButton(_("Next"), self)
self.next_button.setDefault(True)
self.logo = QLabel()
self.please_wait = QLabel(_("Please wait..."))
self.please_wait.setAlignment(Qt.AlignCenter)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
outer_vbox = QVBoxLayout(self)
inner_vbox = QVBoxLayout()
inner_vbox.addWidget(self.title)
inner_vbox.addWidget(self.main_widget)
inner_vbox.addStretch(1)
inner_vbox.addWidget(self.please_wait)
inner_vbox.addStretch(1)
scroll_widget = QWidget()
scroll_widget.setLayout(inner_vbox)
scroll = QScrollArea()
scroll.setWidget(scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
icon_vbox = QVBoxLayout()
icon_vbox.addWidget(self.logo)
icon_vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addLayout(icon_vbox)
hbox.addSpacing(5)
hbox.addWidget(scroll)
hbox.setStretchFactor(scroll, 1)
outer_vbox.addLayout(hbox)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self.set_icon('electrum.png')
self.show()
self.raise_()
self.refresh_gui() # Need for QT on MacOSX. Lame.
def select_storage(self, path, get_wallet_from_daemon):
vbox = QVBoxLayout()
hbox = QHBoxLayout()
hbox.addWidget(QLabel(_('Wallet') + ':'))
self.name_e = QLineEdit()
hbox.addWidget(self.name_e)
button = QPushButton(_('Choose...'))
hbox.addWidget(button)
vbox.addLayout(hbox)
self.msg_label = QLabel('')
vbox.addWidget(self.msg_label)
hbox2 = QHBoxLayout()
self.pw_e = QLineEdit('', self)
self.pw_e.setFixedWidth(150)
self.pw_e.setEchoMode(2)
self.pw_label = QLabel(_('Password') + ':')
hbox2.addWidget(self.pw_label)
hbox2.addWidget(self.pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
self.set_layout(vbox, title=_('Electrum wallet'))
self.storage = WalletStorage(path, manual_upgrades=True)
wallet_folder = os.path.dirname(self.storage.path)
def on_choose():
path, __ = QFileDialog.getOpenFileName(self,
"Select your wallet file",
wallet_folder)
if path:
self.name_e.setText(path)
def on_filename(filename):
path = os.path.join(wallet_folder, filename)
wallet_from_memory = get_wallet_from_daemon(path)
try:
if wallet_from_memory:
self.storage = wallet_from_memory.storage
else:
self.storage = WalletStorage(path, manual_upgrades=True)
self.next_button.setEnabled(True)
except BaseException:
traceback.print_exc(file=sys.stderr)
self.storage = None
self.next_button.setEnabled(False)
if self.storage:
if not self.storage.file_exists():
msg =_("This file does not exist.") + '\n' \
+ _("Press 'Next' to create this wallet, or choose another file.")
pw = False
elif not wallet_from_memory:
if self.storage.is_encrypted_with_user_pw():
msg = _("This file is encrypted with a password.") + '\n' \
+ _('Enter your password or choose another file.')
pw = True
elif self.storage.is_encrypted_with_hw_device():
msg = _("This file is encrypted using a hardware device.") + '\n' \
+ _("Press 'Next' to choose device to decrypt.")
pw = False
else:
msg = _("Press 'Next' to open this wallet.")
pw = False
else:
msg = _("This file is already open in memory.") + "\n" \
+ _("Press 'Next' to create/focus window.")
pw = False
else:
msg = _('Cannot read file')
pw = False
self.msg_label.setText(msg)
if pw:
self.pw_label.show()
self.pw_e.show()
self.pw_e.setFocus()
else:
self.pw_label.hide()
self.pw_e.hide()
button.clicked.connect(on_choose)
self.name_e.textChanged.connect(on_filename)
n = os.path.basename(self.storage.path)
self.name_e.setText(n)
while True:
if self.loop.exec_() != 2: # 2 = next
return
if self.storage.file_exists() and not self.storage.is_encrypted():
break
if not self.storage.file_exists():
break
wallet_from_memory = get_wallet_from_daemon(self.storage.path)
if wallet_from_memory:
return wallet_from_memory
if self.storage.file_exists() and self.storage.is_encrypted():
if self.storage.is_encrypted_with_user_pw():
password = self.pw_e.text()
try:
self.storage.decrypt(password)
break
except InvalidPassword as e:
QMessageBox.information(None, _('Error'), str(e))
continue
except BaseException as e:
traceback.print_exc(file=sys.stdout)
QMessageBox.information(None, _('Error'), str(e))
return
elif self.storage.is_encrypted_with_hw_device():
try:
self.run('choose_hw_device', HWD_SETUP_DECRYPT_WALLET)
except InvalidPassword as e:
QMessageBox.information(
None, _('Error'),
_('Failed to decrypt using this hardware device.')
+ '\n' +
_('If you use a passphrase, make sure it is correct.'
))
self.reset_stack()
return self.run_and_get_wallet(get_wallet_from_daemon)
except BaseException as e:
traceback.print_exc(file=sys.stdout)
QMessageBox.information(None, _('Error'), str(e))
return
if self.storage.is_past_initial_decryption():
break
else:
return
else:
raise Exception('Unexpected encryption version')
return True
def run_and_get_wallet(self):
path = self.storage.path
if self.storage.requires_split():
self.hide()
msg = _(
"The wallet '{}' contains multiple accounts, which are no longer supported since Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?"
).format(path)
if not self.question(msg):
return
file_list = '\n'.join(self.storage.split_accounts())
msg = _('Your accounts have been moved to'
) + ':\n' + file_list + '\n\n' + _(
'Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
return
action = self.storage.get_action()
if action and action not in ('new', 'upgrade_storage'):
self.hide()
msg = _("The file '{}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path)
if not self.question(msg):
if self.question(
_("Do you want to delete '{}'?").format(path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
if action:
# self.wallet is set in run
self.run(action)
return self.wallet
self.wallet = Wallet(self.storage)
return self.wallet
def finished(self):
"""Called in hardware client wrapper, in order to close popups."""
return
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
traceback.print_exception(*exc_info)
self.show_error(str(exc_info[1]))
def set_icon(self, filename):
prior_filename, self.icon_filename = self.icon_filename, filename
self.logo.setPixmap(
QPixmap(icon_path(filename)).scaledToWidth(
60, mode=Qt.SmoothTransformation))
return prior_filename
def set_layout(self, layout, title=None, next_enabled=True):
self.title.setText("<b>%s</b>" % title if title else "")
self.title.setVisible(bool(title))
# Get rid of any prior layout by assigning it to a temporary widget
prior_layout = self.main_widget.layout()
if prior_layout:
QWidget().setLayout(prior_layout)
self.main_widget.setLayout(layout)
self.back_button.setEnabled(True)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self,
layout,
title=None,
raise_on_cancel=True,
next_enabled=True):
self.set_layout(layout, title, next_enabled)
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled
if result == 1:
raise GoBack from None
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
self.app.processEvents()
self.app.processEvents()
def remove_from_recently_open(self, filename):
self.config.remove_from_recently_open(filename)
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self,
header_layout=message,
is_valid=is_valid,
allow_multi=allow_multi)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options):
slayout = SeedLayout(title=message,
is_seed=is_seed,
options=options,
parent=self)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
@wizard_dialog
def add_xpub_dialog(self,
title,
message,
is_valid,
run_next,
allow_multi=False,
show_wif_help=False):
header_layout = QHBoxLayout()
label = WWLabel(message)
label.setMinimumWidth(400)
header_layout.addWidget(label)
if show_wif_help:
header_layout.addWidget(InfoButton(WIF_HELP_TEXT),
alignment=Qt.AlignRight)
return self.text_input(title, header_layout, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d" % index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.'
)
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _(
'Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options)
@wizard_dialog
def confirm_seed_dialog(self, run_next, test):
self.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.'
)
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text):
title = _("Your wallet generation seed is:")
slayout = SeedLayout(seed=seed_text,
title=title,
msg=True,
options=['ext'])
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind, force_disable_encrypt_cb):
playout = PasswordLayout(
msg=msg,
kind=kind,
OK_button=self.next_button,
force_disable_encrypt_cb=force_disable_encrypt_cb)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.new_password(), playout.encrypt_cb.isChecked()
@wizard_dialog
def request_password(self, run_next, force_disable_encrypt_cb=False):
"""Request the user enter a new password and confirm it. Return
the password or None for no password."""
return self.pw_layout(MSG_ENTER_PASSWORD, PW_NEW,
force_disable_encrypt_cb)
@wizard_dialog
def request_storage_encryption(self, run_next):
playout = PasswordLayoutForHW(MSG_HW_STORAGE_ENCRYPTION)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.encrypt_cb.isChecked()
@wizard_dialog
def confirm_dialog(self, title, message, run_next):
self.confirm(message, title)
def confirm(self, message, title):
label = WWLabel(message)
vbox = QVBoxLayout()
vbox.addWidget(label)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self):
self.accept_signal.emit()
def waiting_dialog(self, task, msg, on_finished=None):
label = WWLabel(msg)
vbox = QVBoxLayout()
vbox.addSpacing(100)
label.setMinimumWidth(300)
label.setAlignment(Qt.AlignCenter)
vbox.addWidget(label)
self.set_layout(vbox, next_enabled=False)
self.back_button.setEnabled(False)
t = threading.Thread(target=task)
t.start()
while True:
t.join(1.0 / 60)
if t.is_alive():
self.refresh_gui()
else:
break
if on_finished:
on_finished()
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
clayout = ChoicesLayout(message, c_titles)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
return action
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def choice_and_line_dialog(self,
title: str,
message1: str,
choices: List[Tuple[str, str, str]],
message2: str,
test_text: Callable[[str], int],
run_next,
default_choice_idx: int = 0) -> Tuple[str, str]:
vbox = QVBoxLayout()
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
c_default_text = [x[2] for x in choices]
def on_choice_click(clayout):
idx = clayout.selected_index()
line.setText(c_default_text[idx])
clayout = ChoicesLayout(message1,
c_titles,
on_choice_click,
checked_index=default_choice_idx)
vbox.addLayout(clayout.layout())
vbox.addSpacing(50)
vbox.addWidget(WWLabel(message2))
line = QLineEdit()
def on_text_change(text):
self.next_button.setEnabled(test_text(text))
line.textEdited.connect(on_text_change)
on_choice_click(clayout) # set default text for "line"
vbox.addWidget(line)
self.exec_layout(vbox, title)
choice = c_values[clayout.selected_index()]
return str(line.text()), choice
@wizard_dialog
def line_dialog(self,
run_next,
title,
message,
default,
test,
warning='',
presets=(),
warn_issue4566=False):
vbox = QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
if warn_issue4566:
text_whitespace_normalised = ' '.join(text.split())
warn_issue4566_label.setVisible(
text != text_whitespace_normalised)
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
warn_issue4566_label = WWLabel(MSG_PASSPHRASE_WARN_ISSUE4566)
warn_issue4566_label.setVisible(False)
vbox.addWidget(warn_issue4566_label)
for preset in presets:
button = QPushButton(preset[0])
button.clicked.connect(
lambda __, text=preset[1]: line.setText(text))
button.setMinimumWidth(150)
hbox = QHBoxLayout()
hbox.addWidget(button, alignment=Qt.AlignCenter)
vbox.addLayout(hbox)
self.exec_layout(vbox, title, next_enabled=test(default))
return line.text()
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QVBoxLayout()
layout = SeedLayout(xpub, title=msg, icon=False, for_seed_words=False)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network):
message = _("Electrum communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfill the same purpose only differing in "
"hardware. In most cases you simply want to let Electrum "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_('Cancel'))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
if r == 1:
nlayout = NetworkChoiceLayout(network, self.config, wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
else:
network.auto_connect = True
self.config.set_key('auto_connect', True, True)
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QSlider(Qt.Horizontal, self)
n_edit = QSlider(Qt.Horizontal, self)
n_edit.setMinimum(2)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QLabel()
m_label = QLabel()
grid = QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require {0} signatures').format(m))
cw.set_m(m)
def on_n(n):
n_label.setText(_('From {0} cosigners').format(n))
cw.set_n(n)
m_edit.setMaximum(n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
on_n(2)
on_m(2)
vbox = QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(
WWLabel(
_("Choose the number of signatures needed to unlock funds in your wallet:"
)))
vbox.addLayout(grid)
self.exec_layout(vbox, _("Multi-Signature Wallet"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)
class Kiwoom(QAxWidget):
"""
https://wikidocs.net/5755
키움증권에서로 부터 설치한것은 OCX 방식에 컨포넌트 객체로 설치가 되었다.
응용프로그램에서 키움Open API를 실행 할 수 있게 한거다.
덕분에 제어가 된다.
제어 할수 있는 함수는 PytQ5
시그널 = 요청
키움 API는 싱글데이터 멀티데이터 Key제목을 꼭 보고 가져와야 된다.
"""
def __init__(self):
super().__init__() # 분모에있는 초기값들을 사용하기위해 분모에 있는 init을 사용한다.
print("키움 API 연동")
self.realType = RealType()
############event_loop
self.login_exec = None
######################
############변수모음###
self.account_BankNum = None # 계좌번호
self.sPrevNext = None # 페이지 번호
self.checkBalanceBox = {} # 계좌잔고 계좌종목
self.not_tighteningBox = {} # 미체결
#####################
###########종합일봉리스트########
self.onedaypayListData = [] # 모든 데이터 리스트(중점데이터)
###############################
#############스크린 번호 모음
self.QscreenNumberT = 2000
self.QscreenNumberF = 4000
self.QscreenNumberFi = 5000
self.QscreenNumberS = 6000
self.realTime_start_stop_QscreenNumberO = 1000
############유저 계좌 접근 변수
self.userMoney = None
self.deposit_persent = 0.5
self.deposit_division = 4
###########################
#############이벤트 루프 모음
self.defult_account_info_event_loop = QEventLoop()
###########################################
##############조건선별 file변수###
self.files_stock_diet = {}
###############################
self.get_Ocx_Install()
self.event_slot_List()
self.signal_login_CommConnect()
self.dynamicCall("KOA_Functions(QString, QString)",
"ShowAccountWindow", "")
self.get_account_info()
# 예수금 조회
self.deposit_acount_info()
# 계좌평가잔고내역요청 조회
self.checkBalance_acount_info()
# 종목분석용, 임시용으로 실행
self.calculator_Fn()
# condition_stock.txt에 저장된 종목들을 불러옵니다.
self.read_code()
print("계좌번호 :%s" % self.account_BankNum)
print("페이지 번호 :%s" % self.sPrevNext)
print("계좌잔고 계좌종목 :%s" % self.checkBalanceBox)
print("미체결 :%s" % self.not_tighteningBox)
print("조건선별딕셔너리 :%s" % self.files_stock_diet)
# 각각의 스크린번호 Setting
self.screen_number_settings()
"""
0 : 처음 RealTime 등록했을시
1 : 이후 추가 RealTime 등록시
"""
# realTypeData Test
print(self.realType.REALTYPE['장시작시간']['장운영구분'])
print(self.realType.REALTYPE['장시작시간']['시간'])
self.real_slot_call()
# RealTime_slot_Insert
self.real_slot_event()
def get_Ocx_Install(self):
"""
키움 API를 설치함으로써 응용프로그램들이 우리 window 레지스트리에 저장된다.
레지스트리 편집기에 찾을수 있다.
KHOPENAPI.KHOpenAPICtrl.1 폴더명으로 저장되며 DKHOPENAPI 기본값으로 설정되어있다.
"""
# 응용프로그램을 제어하기 위한 그 설치된 경로를 지정해주어야됨
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
def event_slot_List(self):
# 로그인용 이벤트 슬롯연결
self.OnEventConnect.connect(self.login_slot)
# TR용 이벤트 슬롯연결
self.OnReceiveTrData.connect(self.trData_slot)
# 0503 Testing Error
def real_slot_event(self):
self.OnReceiveRealData.connect(self.realData_slot)
# 로그인 성공시 이벤트
def login_Function(self, nErrCode):
print('로그인처리 %d' % nErrCode)
self.login_exec.exit()
# 키움 로그인 기능창 출력 or 자동설정 (로그인 시도)
def signal_login_CommConnect(self):
"""
로그인 성공시에 대한 이벤트 지정을 안해주면 로그인 성공시에 이벤트값이 없기에
이벤트 핸들러 오류 발생할수 있음
"""
# 네트워크적이거나 다른 서버 응용프로그램에다가 데이터를 전송할수 있음음
self.dynamicCall("CommConnect()")
# QtCore 이벤트 루프 초기화
self.login_exec = QEventLoop()
# 이벤트 루프 지정
self.login_exec.exec_()
# 키움 에러처리 key
def login_slot(self, errCode):
print(errors(errCode))
self.login_exec.exit()
# 계좌번호가져오기
def get_account_info(self):
account_list = self.dynamicCall("GetLoginInfo(QString)", "ACCLIST")
# split : 문자열 자르기 (계좌번호 가져올때 18545454545;45447878785; 이런식으로 출력됨)
self.account_BankNum = account_list.split(';')[0]
# 현재 계좌번호
print('현재 계좌 번호 %s' % self.account_BankNum)
# 예수금 요청내역
def deposit_acount_info(self):
print('예수금 조회 요청')
# 조회 함수입력
self.dynamicCall("SetInputValue(QString, QString)", "계좌번호",
self.account_BankNum) # 계좌번호
self.dynamicCall("SetInputValue(QString, QString)", "비밀번호", "0000")
self.dynamicCall("SetInputValue(QString, QString)", "비밀번호입력매체구분", "00")
self.dynamicCall("SetInputValue(QString, QString)", "조회구분",
'2') #2는 일반조회 3은 추정조회
# 예수금 상세현황요청 조회함수 호출하여 서버전송
self.dynamicCall("CommRqData(QString, QString, QString, QString)",
"예수금상세현황요청", "opw00001", '0', self.QscreenNumberT)
# 다른데이터를 처리하는동안 다른작업을 할수있게 만드는것 이벤트 루프
self.defult_account_info_event_loop
self.defult_account_info_event_loop.exec_()
# 계좌평가잔고내역요청내역, sPrevNext값이 0으로 설정되었기에 싱글데이터값을 요청함함
def checkBalance_acount_info(self, setsPrevNext="0"):
print("계좌평가잔고내역요청")
self.dynamicCall("SetInputValue(QString, QString)", "계좌번호",
self.account_BankNum)
self.dynamicCall("SetInputValue(QString, QString)", "비밀번호", "0000")
self.dynamicCall("SetInputValue(QString, QString)", "비밀번호입력매체구분", "00")
self.dynamicCall("SetInputValue(QString, QString)", "조회구분", '2')
# 계좌평가잔고내역요청 조회함수 호출하여 서버전송
self.dynamicCall("CommRqData(QString, QString, QString, QString)",
"계좌평가잔고내역요청", "opw00018", setsPrevNext,
self.QscreenNumberF)
self.defult_account_info_event_loop.exec_()
# 미체결요청
def not_tightening_acount_info(self):
print("미체결요청")
# 조회하고 싶은것만 받아오게 할수있다.
self.dynamicCall("SetInputValue(QString, QString)", "계좌번호",
self.account_BankNum)
self.dynamicCall("SetInputValue(QString, QString)", "매매구분", 0)
self.dynamicCall("SetInputValue(QString, QString)", "체결구분", 1)
# Kiwoom서버에게 해당 데이터값 요청
self.dynamicCall("CommRqData(QString, QString, QString, QString)",
"미체결요청", "opt10075", 0, self.QscreenNumber)
self.defult_account_info_event_loop.exec_()
# 주식일봉차트조회요청
def request_to_check_the_stock_pay_chart(self,
eventCode=None,
date=None,
setsPrevNext=0):
print("주식일봉차트조회")
"""
위 네트워크적인 프로세스를 멈추진 않고 대신에 이상태에서 다음 코드실행하기 전에 4초 지연합니다. (보안상 3.5초 이상 지연)
너무 느리지만 이점을 보완할 방법이 딱히 없다.
그리고 키움은 매일 새벽마다 업데이트를 한다.
"""
QTest.qWait(4000)
self.dynamicCall("SetInputValue(QString, QString)", "종목코드", eventCode)
self.dynamicCall("SetInputValue(QString, QString)", "수정주가구분", "1")
if date != None:
self.dynamicCall("SetInputValue(QString, QString)", "기준일자", date)
self.dynamicCall("CommRqData(QString, QString, int, QString)",
"주식일봉차트조회", "opt10081", setsPrevNext,
self.QscreenNumberF) #TR서버로 전송
self.defult_account_info_event_loop.exec_()
def get_code_list_by_market(self, market_code):
'''
종목코드들 반환
:param market_code:
:return:
'''
code_list = self.dynamicCall("GetCodeListByMarket(QString)",
market_code)
code_list = code_list.split(";")[:-1]
return code_list
def calculator_Fn(self):
code_list = self.get_code_list_by_market("10")
print("코스닥 갯수 %s", len(code_list))
# 데이터가 너무 많아 1449이기에 3개로 줄여서 테스팅실시함.
for index, code in enumerate(code_list[0:3]):
"""
추가 : 스크린번호를 한번이라도 요청하면 그룹이 만들어 진것이다.
그래서 끊어주는건 개인의 선택이다.
"""
# 화면번호 끊기
self.dynamicCall("DisconnectRealData(QString)",
self.QscreenNumberF)
print("%s / %s : Kosdaq code_list : %s" %
(index + 1, len(code_list), code))
self.request_to_check_the_stock_pay_chart(eventCode=code)
# Tr슬롯
def trData_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):
'''
TR 요청을 받는 구역 (슬롯)
:param sScrNo: 화면번호
:param sRQName: 사용자 구분명
:param sTrCode: TR이름
:param sRecordName: 레코드이름
:param sPrevNext: 연속조회 유무를 판단하는 값 0: 연속(추가조회)데이터 없음, 2:연속(추가조회) 데이터 있음 (다음페이지가 있는지)
sPrevNext는 다음페이지가 활성화가 되면 2로 반환됩니다. 만약 다음페이지가 비활성화 상태라면 첫페이지인 0 or ""이 반환됩니다.
다음페이지 활성화조건은 계좌평가잔고내역요청에 종목코드가 20개 초과일시 생성되어야합니다.
:return:
'''
print(sTrCode)
if (sRQName == "예수금상세현황요청"):
"""
GetCommData
- OnReceiveTRData()이벤트가 발생될때 수신한 데이터를 얻어오는 함수입니다.
이 함수는 OnReceiveTRData()이벤트가 발생될때 그 안에서 사용해야 합니다.
"""
# TR이름, 레코드이름인데 사용자 구분명으로 대체함;, 0은 아직도 잘 모르겠다, 필드항목이름이 예수금임
deposit = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, 0, "예수금")
print(deposit)
print("예수금 : %s" % deposit)
print("예수금 형변환 : %s" % int(deposit))
###### 유저 한번 구입할때 조절하는 수치 (고객이 직접조절하게 GUI 개선해야됨)
self.userMoney = int(deposit) * self.deposit_persent
self.userMoney = self.userMoney / self.deposit_division
stock_evidence_cash = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, 0, "주식증거금현금")
print("주식증거금현금 : %s" % stock_evidence_cash)
print("주식증거금현금 형변환 : %s" % int(stock_evidence_cash))
ture_deposit = self.dynamicCall(
"GetCommData(QString, QString, QString, QString)", sTrCode,
sRQName, 0, "출금가능금액")
print("출금가능금액 : %s" % ture_deposit)
print("출금가능금액 형변환 : %d" % int(ture_deposit))
self.defult_account_info_event_loop.exit()
elif (sRQName == "계좌평가잔고내역요청"):
# 계좌평가잔고내역요청_총매입금액
checkBalance_total_purchase_amount = self.dynamicCall(
"GetCommData(QString, QString, QString, QString)", sTrCode,
sRQName, 0, "총매입금액")
print('총매입금액 : %s' % checkBalance_total_purchase_amount)
print('총매입금액 형변환 : %s' % int(checkBalance_total_purchase_amount))
# 계좌평가잔고내역요청_총평가금액
checkBalance_total_evaluation_amount = self.dynamicCall(
"GetCommData(QString, QString, QString, QString)", sTrCode,
sRQName, 0, "총평가금액")
print('총평가금액 : %s' % checkBalance_total_evaluation_amount)
print('총평가금액 형변환 : %s' % int(checkBalance_total_evaluation_amount))
# 계좌평가잔고내역요청_총수익률
checkBalance_yield = self.dynamicCall(
"GetCommData(QString, QString, QString, QString)", sTrCode,
sRQName, 0, "총수익률(%)")
print('총수익률(%%) : %s' % checkBalance_yield)
print('총수익률(%%) 형변환 : %s' % float(checkBalance_yield), '%')
# 현재 주식 매매 카운트 (GetRepeatCnt를 사용하면 카운트가 멀티데이터로 됨)
cint = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode,
sRQName)
# 종목이 몇개 인지 뽑아보기 위해서 함
# 600일의 관한 일봉 데이터
for i in range(cint):
balance_id = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "종목번호")
balance_item_nm = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "종목명")
balance_retained_quantity = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "보유수량")
balance_purchase_price = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "매입가")
balance_yield = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "수익률(%)")
balance_current_price = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "현재가")
balance_purchase_amount = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "매입금액")
balance_tradeable_quantity = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "매매가능수량")
# checkBalance 안에 cint이 있다면
if (cint in self.checkBalanceBox):
pass
else:
# self.checkBalanceBox[balance_id] = {} 이렇게 사용해도됨
balance_id = balance_id.strip()[1:]
self.checkBalanceBox.update({balance_id: {}})
# strip(공백없애기) 하지만 [1:] 이면 2번째요소(인덱스자리 1)부터 시작한다는뜻
balance_item_nm = balance_item_nm.strip()
balance_retained_quantity = int(
balance_retained_quantity.strip())
balance_purchase_price = int(balance_purchase_price.strip())
balance_yield = float(balance_yield.strip())
balance_current_price = int(balance_current_price.strip())
balance_purchase_amount = int(balance_purchase_amount.strip())
balance_tradeable_quantity = int(
balance_tradeable_quantity.strip())
checkBalanceBoxSkip = self.checkBalanceBox[balance_id]
# 종목 코드 해당 id 딕셔너리 안에 그 안에 딕셔너리 추가함
checkBalanceBoxSkip.update({"종목명": balance_item_nm})
checkBalanceBoxSkip.update({"보유수량": balance_retained_quantity})
checkBalanceBoxSkip.update({"매입가": balance_purchase_price})
checkBalanceBoxSkip.update({"수익률(%)": balance_yield})
checkBalanceBoxSkip.update({"현재가": balance_current_price})
checkBalanceBoxSkip.update({"매입금액": balance_purchase_amount})
checkBalanceBoxSkip.update(
{"매매가능수량": balance_tradeable_quantity})
print(
"종목코드: %s - 종목명: %s - 보유수량: %s - 매입가:%s - 수익률: %s - 현재가: %s"
% (balance_id, balance_item_nm, balance_retained_quantity,
balance_purchase_price, balance_yield,
balance_current_price))
print("계좌 가지고 있는종목 : %d" % len(self.checkBalanceBox))
if (sPrevNext == "2"):
self.checkBalance_acount_info(setsPrevNext="2")
else:
self.defult_account_info_event_loop.exit()
print(self.checkBalanceBox.keys())
print(self.checkBalanceBox.values())
elif (sRQName == "미체결요청"):
# 미체결요청에 GetRepeatCnt은 최대 100Count
not_tightening_List = self.dynamicCall(
"GetRepeatCnt(QStirng, QString)", sTrCode, sRQName)
for i in range(not_tightening_List):
not_tightening_id = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "종목코드")
not_tightening_nm = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "종목명")
# -매도 +매수, -매도정리
not_tightening_order_sortation = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "주문구분")
# 접수 -> 확인 -> 체결
not_tightening_order_state = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "주문상태")
not_tightening_order_quantity = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "주문수량")
not_tightening_order_price = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "주문가격")
# 기본키
not_tightening_order_no = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "주문번호")
not_tightening_quantity = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "미체결수량")
tightening_quantity = self.dynamicCall(
"GetCommData(QString QString int QString)", sTrCode,
sRQName, i, "체결량")
not_tightening_id = not_tightening_id.strip()
not_tightening_nm = not_tightening_nm.strip()
# 첫문자 + or - 를 제거
not_tightening_order_sortation = not_tightening_order_sortation.strip(
).lstrip('+').lstrip('-')
not_tightening_order_state = not_tightening_order_state.strip()
not_tightening_order_quantity = int(
not_tightening_order_quantity.strip())
not_tightening_order_price = int(
not_tightening_order_price.strip())
not_tightening_order_no = int(not_tightening_order_no.strip())
not_tightening_quantity = int(not_tightening_quantity.strip())
tightening_quantity = int(tightening_quantity.strip())
if not_tightening_order_no in self.not_tighteningBox:
pass
else:
self.not_tighteningBox.update(
{not_tightening_order_no, {}})
not_tighteningBox_skip = self.not_tighteningBox[
not_tightening_order_no]
not_tighteningBox_skip.update({"종목코드", not_tightening_id})
not_tighteningBox_skip.update({"종목명", not_tightening_nm})
not_tighteningBox_skip.update(
{"주문구분", not_tightening_order_sortation})
not_tighteningBox_skip.update(
{"주문상태", not_tightening_order_state})
not_tighteningBox_skip.update(
{"주문수량", not_tightening_order_quantity})
not_tighteningBox_skip.update(
{"주문가격", not_tightening_order_price})
not_tighteningBox_skip.update(
{"미체결수량", not_tightening_quantity})
not_tighteningBox_skip.update({"체결량", tightening_quantity})
print("미체결 관련 데이터 %s" % self.not_tighteningBox)
elif "주식일봉차트조회" == sRQName:
"""
장기적인 투자면 이방법이 편하지만 단기적으로 해야된다면 이방법은 추천드리지 않는다.
문제점 방대한 데이터를 전부 instoll하기엔 너무 오래걸린다.
그렇기에 특정 종목코드를 지정하거나 데이터수를 리미트를 걸어서 가져온다.
"""
# 싱글데이터
code = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, 0, "종목코드")
code = code.strip()
print("%s 일봉데이터 요청" % code)
# 최대 호출 카운트
cint = self.dynamicCall("GetRepeatCnt(QString, QString)", sTrCode,
sRQName)
print("데이터 일수 %s" % cint)
# 하루치들의 데이터들 600번 반복
for i in range(cint):
data = []
current_price = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "현재가") #종가
transaction_volume = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "거래량")
transaction_amount = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "거래대금")
date = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "일자")
market_value = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "시가")
high_value = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "고가")
low_value = self.dynamicCall(
"GetCommData(QString, QString, int, QString)", sTrCode,
sRQName, i, "저가")
data.append(current_price.strip())
data.append(transaction_volume.strip())
data.append(transaction_amount.strip())
data.append(date.strip())
data.append(market_value.strip())
data.append(high_value.strip())
data.append(low_value.strip())
self.onedaypayListData.append(data.copy())
print("현재 일봉 데이터량 %s" % len(self.onedaypayListData))
#print("현재 데이터 리스트 %s" % self.onedaypayListData)
if sPrevNext == "2":
self.request_to_check_the_stock_pay_chart(
eventCode=code, setsPrevNext=sPrevNext)
else:
print("총 일수 %s" % len(self.onedaypayListData))
pass_success = None
# 아직은 분석을 더해야됨 자세히 분석하지 못하였음
# 120일 이평선을 그릴만큼의 데이터가 있는지 체크해야됨 (120일 이평선을 보기위한 행위)
if self.onedaypayListData == None or len(
self.onedaypayListData) < 120:
pass_success = False
# 120일 이상 일시
else:
total_price = 0
# 리스트 슬라이싱
for value in self.onedaypayListData[:120]:
# 120일차에 1일현재가들을 전부 더한 값 total_price
total_price += int(value[0])
# 120일치 이평선 평균
average_horizontal = total_price / 120
# 오늘자 주기기 120일 이평선에 걸쳐있는지 확인
bottom_stock_price = False
check_price = None
# 현재 봉(양봉,음봉 구별은 아직안함)이 평균이평선 안에 속해있을때
if int(
self.onedaypayListData[0][6]
) <= average_horizontal and average_horizontal <= int(
self.onedaypayListData[0][5]):
print("오늘 주가 120이평선 확인")
# 이평선이 저가가 평균값보다 낮고 고가가 평균값보다 높아야합니다.
bottom_stock_price = True
price_top_move = False
# 현재 고가 가격
check_price = int(self.onedaypayListData[0][5])
# 과거 일봉들이 120일 이평선보다 밑에 있는지 확인
# 그렇게 확인을 하다가 일봉이 120일 이평선보다 위에 있으면 계산 진행
if bottom_stock_price == True:
average_horizontal = 0
price_top_move = False
index = 1
while True:
# 120일치가 있는지 계속 확인
if len(self.onedaypayListData[index:]) < 120:
print("120일치가 없음!")
break
# 120일치 이평선이 있다는것이 확인된후 다시 구하기 위해 0 설정
total_price = 0
# 이해불능?
for value in self.onedaypayListData[index:120 +
index]:
# 전날 120일차 이평선 현 시점 평균값을 구하기위한 전날 120일 총 현재가
total_price += int(value[0])
# 전날 평균 이평선 값
move_average_horizontal = total_price / 120
# 전날 평균 이평선값이 전날 고가보다 낮고 이 상태로 20일보다 작아야 합니다.
if move_average_horizontal <= int(
self.onedaypayListData[index]
[5]) and index <= 20:
print(
"20일 동안 주기가 120일 이평선과 같거나 위에 있으면 조건 통과 못함"
)
price_top_move = False
break
# 전날 평균 이평선 값이 전날 저가보다 더 작고 20일차보다 커야됩니다.
elif int(
self.onedaypayListData[index][6]
) > move_average_horizontal and index > 20:
print("120일 이평선 위에 있는 일봉 확인됨")
price_top_move = True
# 전날 저가 가격
prev_price = int(
self.onedaypayListData[index][6])
break
else:
index += 1
# 해당 부분 이평선이 가장 최근 일자의 이평선보다 가격이 낮은지 확인 하면서 현재 저가값이 전날 저가값보다
# 더 커야됩니다.
if price_top_move == True:
if average_horizontal > move_average_horizontal and check_price > prev_price:
print(
"포착된 이평선의 가격이 오늘자(최근일자) 이평선 가격보다 낮은 것 확인됨"
)
print(
"포착된 부분의 일봉 저가가 오늘자 일봉의 고가보다 낮은지 확인됨")
pass_success = True
if pass_success == True:
print("조건부 통과됨")
# 종목코드로 코드네임을 가져옵니다.
code_nm = self.dynamicCall(
"GetMasterCodeName(QString)", code)
# open이라는 파이썬 내장함수가 있습니다.
files_direction = open("files/condition_stock.txt",
"a",
encoding="utf8")
files_direction.write(
"%s\t%s\t%s\n" %
(code, code_nm, str(self.onedaypayListData[0][0])))
files_direction.close()
elif pass_success == False:
print("조건부 통과 못함")
self.onedaypayListData.clear()
self.defult_account_info_event_loop.exit()
self.defult_account_info_event_loop.exit()
def read_code(self):
if os.path.exists("S:/Python/Project_Yanajo/files/condition_stock.txt"
): # 있으면 Ture 없으면 False
# r = reading
files_direction = open(
"S:/Python/Project_Yanajo/files/condition_stock.txt",
"r",
encoding="utf8")
lines = files_direction.readlines()
for line in lines:
if line != "":
line_list = line.split("\t")
stock_code = line_list[0]
stock_name = line_list[1]
stock_price = int(line_list[2].split("\n")[0])
stock_price = abs(stock_price)
# {"2027": {"종목명":"삼성 ....}....}]
self.files_stock_diet.update(
{stock_code: {
"종목명": stock_name,
"현재가": stock_price
}})
# 메모리를 잡아먹지 않기위해 file을 닫아줍니다.
files_direction.close()
print(self.files_stock_diet)
def screen_number_settings(self):
# 겹치는 종목들을 알기위함
screemOvering = []
# 계좌잔고내역 종목 getter
for getValueList in self.checkBalanceBox.keys():
if getValueList not in screemOvering:
# 0429(58-24:19)
# 겹치는종목이 없다면은 해당 종목을 추가해줍니다.
screemOvering.append(getValueList)
# 미체결에 있는 종목들
for getValueList in self.not_tighteningBox.keys():
code = self.not_tighteningBox[getValueList]['종목코드']
if code not in getValueList:
screemOvering.append(code)
# 리스트에 담겨있는 목록들
for getValueList in self.files_stock_diet.keys():
if getValueList not in screemOvering:
screemOvering.append(getValueList)
# 스크린번호 할당
count = 0
for getValueList in screemOvering:
temp_screen = int(self.QscreenNumberFi)
meme_screen = int(self.QscreenNumberS)
# 100번째에 끊음
if (count % 100) == 0:
temp_screen += 1 #스크린 번호 하나당 데이터(종목코드) 100개를 넣어줌
self.QscreenNumberFi = str(temp_screen)
meme_screen += 1 # 스크린 번호 하나당 데이터(종목코드) 100개를 넣어줌
self.QscreenNumberS = str(meme_screen)
# 포함될때
if getValueList in self.files_stock_diet.keys():
# 각 종목들 한번에 모으기
self.files_stock_diet[getValueList].update({
"스크린번호":
self.QscreenNumberFi,
"주문용스크린번호":
self.QscreenNumberS
})
# 포함안될때
elif getValueList not in self.files_stock_diet.keys():
# 각 종목들 한번에 모으기
self.files_stock_diet.update({
getValueList: {
"스크린번호": self.QscreenNumberFi,
"주문용스크린번호": self.QscreenNumberS
}
})
count += 1
print("files Array : %s" % self.files_stock_diet)
"""
programer -> (삼성) -> 증권서버
프로그래머가 삼성을 등록해놓음
미리 이벤트랑 슬롯이랑 연결해놓은 구간이있습니다.
그전에는 OnReceiveTrData를 통해서 슬롯과 연결했지만
이번에는 OnReceiveRealData.connet(실시간수신슬롯)을 등록합니다.
삼성이라는 데이터가 바뀌면은 바뀔때마다 서버에서 슬롯을 던져줍니다.
여러가지를 불러오면서 files_stock_diet를 실시간 업데이트를 해줄겁니다.
그리고 매수매도를 여기서 확인을 합니다.
"""
# 0503 Testing Error
def real_slot_call(self):
self.dynamicCall("SetRealReg(QString, QString, QString, QString",
self.realTime_start_stop_QscreenNumberO, '',
self.realType.REALTYPE['장시작시간']['장운영구분'], "0")
self.defult_account_info_event_loop.exec_()
# 0503 Error : 실행안됨
def realData_slot(self, sCode, sRealType, sRealData):
"""
:param sCode: 종목코드
:param sRealType: 실시간타입
:param sRealData: 실시간 데이터 전문 (사용불능)
:return: respons Data
"""
print("RealTimeCode : %s" % sCode)
print("RealTimeType : %s" % sRealType)
print("RealTimeData : %s" % sRealData)
if sRealType == "장시작시간":
fid = self.realType.REALTYPE[sRealType]['장운영구분']
values = self.dynamicCall("GetCommRealData(QString, QString)",
sCode, fid)
# Test
print(values)
if values == 0:
print('현재 시장이 닫은 시간이지만 시작 전 상태 입니다.')
elif values == 3:
print('현재 시장 운영중 입니다.')
elif values == 2:
print('현재 장 종료 10분 전 동시 호가입니다.')
elif values == 4:
print('현재 시장이 종료 되어 있습니다..')
class InstallWizard(QtWidgets.QDialog, MessageBoxMixin, BaseWizard):
accept_signal = pyqtSignal()
def __init__(self, config, app, plugins):
BaseWizard.__init__(self, config)
QtWidgets.QDialog.__init__(self, None)
self.setWindowTitle(f'{PROJECT_NAME} - ' + _('Install Wizard'))
self.app = app
self.config = config
# Set for base base class
self.plugins = plugins
self.setMinimumSize(600, 400)
self.accept_signal.connect(self.accept)
self.title = QtWidgets.QLabel()
self.main_widget = QtWidgets.QWidget()
self.back_button = QtWidgets.QPushButton(_("Back"), self)
self.back_button.setText(_('Back') if self.can_go_back() else _('Cancel'))
self.next_button = QtWidgets.QPushButton(_("Next"), self)
self.next_button.setDefault(True)
self.logo = QtWidgets.QLabel()
self.please_wait = QtWidgets.QLabel(_("Please wait..."))
self.please_wait.setAlignment(Qt.AlignCenter)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
outer_vbox = QtWidgets.QVBoxLayout(self)
inner_vbox = QtWidgets.QVBoxLayout()
inner_vbox.addWidget(self.title)
inner_vbox.addWidget(self.main_widget)
inner_vbox.addStretch(1)
inner_vbox.addWidget(self.please_wait)
inner_vbox.addStretch(1)
scroll_widget = QtWidgets.QWidget()
scroll_widget.setLayout(inner_vbox)
scroll = QtWidgets.QScrollArea()
scroll.setWidget(scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
icon_vbox = QtWidgets.QVBoxLayout()
icon_vbox.addWidget(self.logo)
icon_vbox.addStretch(1)
hbox = QtWidgets.QHBoxLayout()
hbox.addLayout(icon_vbox)
hbox.addSpacing(5)
hbox.addWidget(scroll)
hbox.setStretchFactor(scroll, 1)
outer_vbox.addLayout(hbox)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self.set_icon(':icons/electrumABC.svg')
self.show()
self.raise_()
# Track object lifecycle
finalization_print_error(self)
def select_storage(
self, path, get_wallet_from_daemon
) -> Tuple[str, Optional[WalletStorage]]:
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(
QtWidgets.QLabel(
_('Create a new wallet or load an existing wallet from file.'))
)
vbox.addSpacing(20)
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(QtWidgets.QLabel(_('Wallet name') + ':'))
self.name_e = QtWidgets.QLineEdit()
self.name_e.setToolTip(
_("Enter a wallet name (new or existing), or the path to a wallet file.")
)
hbox.addWidget(self.name_e)
button = QtWidgets.QPushButton(_('Load...'))
button.setToolTip(_("Open a file selection dialog to load a wallet file."))
hbox.addWidget(button)
vbox.addLayout(hbox)
vbox.addSpacing(20)
self.msg_label = QtWidgets.QLabel('')
vbox.addWidget(self.msg_label)
hbox2 = QtWidgets.QHBoxLayout()
self.pw_e = PasswordLineEdit('', self)
self.pw_e.setFixedWidth(17 * char_width_in_lineedit())
self.pw_label = QtWidgets.QLabel(_('Password') + ':')
hbox2.addWidget(self.pw_label)
hbox2.addWidget(self.pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
self.set_layout(vbox, title=_(f'{PROJECT_NAME} wallet'))
self.temp_storage = WalletStorage(path, manual_upgrades=True)
wallet_folder = os.path.dirname(self.temp_storage.path)
def on_choose():
path, __ = QtWidgets.QFileDialog.getOpenFileName(self, "Select your wallet file", wallet_folder)
if path:
self.name_e.setText(path)
def on_filename(filename):
path = os.path.join(wallet_folder, filename)
wallet_from_memory = get_wallet_from_daemon(path)
try:
if wallet_from_memory:
self.temp_storage = wallet_from_memory.storage
else:
self.temp_storage = WalletStorage(path, manual_upgrades=True)
self.next_button.setEnabled(True)
except IOError:
self.temp_storage = None
self.next_button.setEnabled(False)
if self.temp_storage:
if not self.temp_storage.file_exists():
msg =_("This file does not exist.") + '\n' \
+ _("Press 'Next' to create this wallet, or choose another file.")
pw = False
elif not wallet_from_memory:
if self.temp_storage.is_encrypted_with_user_pw():
msg = _("This file is encrypted with a password.") + '\n' \
+ _('Enter your password or choose another file.')
pw = True
elif self.temp_storage.is_encrypted_with_hw_device():
msg = _("This file is encrypted using a hardware device.") + '\n' \
+ _("Press 'Next' to choose device to decrypt.")
pw = False
else:
msg = _("Press 'Next' to open this wallet.")
pw = False
else:
msg = _("This file is already open in memory.") + "\n" \
+ _("Press 'Next' to create/focus window.")
pw = False
else:
msg = _('Cannot read file')
pw = False
self.msg_label.setText(msg)
if pw:
self.pw_label.show()
self.pw_e.show()
self.pw_e.setFocus()
else:
self.pw_label.hide()
self.pw_e.hide()
button.clicked.connect(on_choose)
self.name_e.textChanged.connect(on_filename)
n = os.path.basename(self.temp_storage.path)
self.name_e.setText(n)
while True:
if self.loop.exec_() != 2: # 2 = next
raise UserCancelled
if self.temp_storage.file_exists() and not self.temp_storage.is_encrypted():
break
if not self.temp_storage.file_exists():
break
wallet_from_memory = get_wallet_from_daemon(self.temp_storage.path)
if wallet_from_memory:
raise WalletAlreadyOpenInMemory(wallet_from_memory)
if self.temp_storage.file_exists() and self.temp_storage.is_encrypted():
if self.temp_storage.is_encrypted_with_user_pw():
password = self.pw_e.text()
try:
self.temp_storage.decrypt(password)
break
except InvalidPassword as e:
QtWidgets.QMessageBox.information(None, _('Error'), str(e))
continue
except BaseException as e:
traceback.print_exc(file=sys.stdout)
QtWidgets.QMessageBox.information(None, _('Error'), str(e))
raise UserCancelled()
elif self.temp_storage.is_encrypted_with_hw_device():
try:
self.run(
'choose_hw_device',
HWD_SETUP_DECRYPT_WALLET,
storage=self.temp_storage
)
except InvalidPassword as e:
QtWidgets.QMessageBox.information(
None, _('Error'),
_('Failed to decrypt using this hardware device.') + '\n' +
_('If you use a passphrase, make sure it is correct.'))
self.reset_stack()
return self.select_storage(path, get_wallet_from_daemon)
except BaseException as e:
traceback.print_exc(file=sys.stdout)
QtWidgets.QMessageBox.information(None, _('Error'), str(e))
raise UserCancelled()
if self.temp_storage.is_past_initial_decryption():
break
else:
raise UserCancelled()
else:
raise Exception('Unexpected encryption version')
return self.temp_storage.path, (self.temp_storage if self.temp_storage.file_exists() else None)
def run_upgrades(self, storage):
path = storage.path
if storage.requires_split():
self.hide()
msg = _("The wallet '{}' contains multiple accounts, which are no longer supported since Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?").format(path)
if not self.question(msg):
return
file_list = '\n'.join(storage.split_accounts())
msg = _('Your accounts have been moved to') + ':\n' + file_list + '\n\n'+ _('Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
# raise now, to avoid having the old storage opened
raise UserCancelled()
action = storage.get_action()
if action and storage.requires_upgrade():
raise WalletFileException('Incomplete wallet files cannot be upgraded.')
if action:
self.hide()
msg = _("The file '{}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path)
if not self.question(msg):
if self.question(_("Do you want to delete '{}'?").format(path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
self.data = storage.db.data
self.run(action)
for k, v in self.data.items():
storage.put(k, v)
storage.write()
return
if storage.requires_upgrade():
self.upgrade_storage(storage)
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
traceback.print_exception(*exc_info)
self.show_error(str(exc_info[1]))
def set_icon(self, filename):
prior_filename, self.icon_filename = self.icon_filename, filename
self.logo.setPixmap(QIcon(filename).pixmap(60))
return prior_filename
def set_layout(self, layout, title=None, next_enabled=True):
self.title.setText("<b>%s</b>"%title if title else "")
self.title.setVisible(bool(title))
# Get rid of any prior layout by assigning it to a temporary widget
prior_layout = self.main_widget.layout()
if prior_layout:
QtWidgets.QWidget().setLayout(prior_layout)
self.main_widget.setLayout(layout)
self.back_button.setEnabled(True)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self, layout, title=None, raise_on_cancel=True,
next_enabled=True):
self.set_layout(layout, title, next_enabled)
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled
if result == 1:
raise GoBack from None
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
self.app.processEvents()
self.app.processEvents()
def remove_from_recently_open(self, filename):
self.config.remove_from_recently_open(filename)
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self, title=message, is_valid=is_valid,
allow_multi=allow_multi)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options):
slayout = SeedLayout(title=message, is_seed=is_seed, options=options, parent=self, editable=True)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
def bip38_prompt_for_pw(self, bip38_keys):
''' Reimplemented from basewizard superclass. Expected to return the pw
dict or None. '''
d = Bip38Importer(bip38_keys, parent=self.top_level_window())
res = d.exec_()
d.setParent(None) # python GC quicker if this happens
return d.decoded_keys # dict will be empty if user cancelled
@wizard_dialog
def add_xpub_dialog(self, title, message, is_valid, run_next, allow_multi=False):
return self.text_input(title, message, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d"%index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.')
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _('Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options)
@wizard_dialog
def confirm_seed_dialog(self, run_next, test):
self.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.')
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text, editable=True):
title = _("Your wallet generation seed is:")
slayout = SeedLayout(seed=seed_text, title=title, msg=True, options=['ext'], editable=False)
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind, force_disable_encrypt_cb):
playout = PasswordLayout(msg=msg, kind=kind, OK_button=self.next_button,
force_disable_encrypt_cb=force_disable_encrypt_cb)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.new_password(), playout.encrypt_cb.isChecked()
@wizard_dialog
def request_password(self, run_next, force_disable_encrypt_cb=False):
"""Request the user enter a new password and confirm it. Return
the password or None for no password. Note that this dialog screen
cannot go back, and instead the user can only cancel."""
return self.pw_layout(MSG_ENTER_PASSWORD, PW_NEW, force_disable_encrypt_cb)
@wizard_dialog
def request_storage_encryption(self, run_next):
playout = PasswordLayoutForHW(MSG_HW_STORAGE_ENCRYPTION)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.encrypt_cb.isChecked()
@staticmethod
def _add_extra_button_to_layout(extra_button, layout):
if (not isinstance(extra_button, (list, tuple))
or not len(extra_button) == 2):
return
but_title, but_action = extra_button
hbox = QtWidgets.QHBoxLayout()
hbox.setContentsMargins(12,24,12,12)
but = QtWidgets.QPushButton(but_title)
hbox.addStretch(1)
hbox.addWidget(but)
layout.addLayout(hbox)
but.clicked.connect(but_action)
@wizard_dialog
def confirm_dialog(self, title, message, run_next, extra_button=None):
self.confirm(message, title, extra_button=extra_button)
def confirm(self, message, title, extra_button=None):
label = WWLabel(message)
textInteractionFlags = (Qt.LinksAccessibleByMouse
| Qt.TextSelectableByMouse
| Qt.TextSelectableByKeyboard
| Qt.LinksAccessibleByKeyboard)
label.setTextInteractionFlags(textInteractionFlags)
label.setOpenExternalLinks(True)
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(label)
if extra_button:
self._add_extra_button_to_layout(extra_button, vbox)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self, **kwargs):
self.accept_signal.emit()
def run_task_without_blocking_gui(self, task, *, msg=None):
assert self.gui_thread == threading.current_thread(), 'must be called from GUI thread'
if msg is None:
msg = _("Please wait...")
exc: Optional[Exception] = None
res = None
def task_wrapper():
nonlocal exc
nonlocal res
try:
task()
except Exception as e:
exc = e
self.waiting_dialog(task_wrapper, msg=msg)
if exc is None:
return res
else:
raise exc
def waiting_dialog(self, task, msg, on_finished=None):
label = WWLabel(msg)
vbox = QtWidgets.QVBoxLayout()
vbox.addSpacing(100)
label.setMinimumWidth(300)
label.setAlignment(Qt.AlignCenter)
vbox.addWidget(label)
self.set_layout(vbox, next_enabled=False)
self.back_button.setEnabled(False)
t = threading.Thread(target=task)
t.start()
while True:
t.join(1.0/60)
if t.is_alive():
self.refresh_gui()
else:
break
if on_finished:
on_finished()
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next, extra_button=None):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
clayout = ChoicesLayout(message, c_titles)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(clayout.layout())
if extra_button:
self._add_extra_button_to_layout(extra_button, vbox)
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
return action
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QtWidgets.QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def line_dialog(self, run_next, title, message, default, test, warning=''):
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QtWidgets.QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
self.exec_layout(vbox, title, next_enabled=test(default))
return ' '.join(line.text().split())
@wizard_dialog
def derivation_path_dialog(self, run_next, title, message, default, test, warning='', seed='', scannable=False):
def on_derivation_scan(derivation_line, seed):
derivation_scan_dialog = DerivationDialog(self, seed, DerivationPathScanner.DERIVATION_PATHS)
destroyed_print_error(derivation_scan_dialog)
selected_path = derivation_scan_dialog.get_selected_path()
if selected_path:
derivation_line.setText(selected_path)
derivation_scan_dialog.deleteLater()
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QtWidgets.QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
if scannable:
hbox = QtWidgets.QHBoxLayout()
hbox.setContentsMargins(12,24,12,12)
but = QtWidgets.QPushButton(_("Scan Derivation Paths..."))
hbox.addStretch(1)
hbox.addWidget(but)
vbox.addLayout(hbox)
but.clicked.connect(lambda: on_derivation_scan(line, seed))
self.exec_layout(vbox, title, next_enabled=test(default))
return ' '.join(line.text().split())
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QtWidgets.QVBoxLayout()
layout = SeedLayout(xpub, title=msg, icon=False)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network):
message = _(
f"{PROJECT_NAME} communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfil the same purpose only differing in "
f"hardware. In most cases you simply want to let {PROJECT_NAME} "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_('Cancel'))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
network.auto_connect = (r == 0)
self.config.set_key('auto_connect', network.auto_connect, True)
if r == 1:
nlayout = NetworkChoiceLayout(self, network, self.config, wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QtWidgets.QSlider(Qt.Horizontal, self)
n_edit = QtWidgets.QSlider(Qt.Horizontal, self)
n_edit.setMinimum(1)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QtWidgets.QLabel()
m_label = QtWidgets.QLabel()
grid = QtWidgets.QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require %d signatures')%m)
cw.set_m(m)
def on_n(n):
n_label.setText(_('From %d cosigners')%n)
cw.set_n(n)
m_edit.setMaximum(n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
on_n(2)
on_m(2)
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(WWLabel(_("Choose the number of signatures needed to unlock funds in your wallet:")))
vbox.addLayout(grid)
self.exec_layout(vbox, _("Multi-Signature Wallet"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)
linux_hw_wallet_support_dialog = None
def on_hw_wallet_support(self):
''' Overrides base wizard's noop impl. '''
if sys.platform.startswith("linux"):
if self.linux_hw_wallet_support_dialog:
self.linux_hw_wallet_support_dialog.raise_()
return
# NB: this should only be imported from Linux
from . import udev_installer
self.linux_hw_wallet_support_dialog = udev_installer.InstallHardwareWalletSupportDialog(self.top_level_window(), self.plugins)
self.linux_hw_wallet_support_dialog.exec_()
self.linux_hw_wallet_support_dialog.setParent(None)
self.linux_hw_wallet_support_dialog = None
else:
self.show_error("Linux only facility. FIXME!")
class PasswordChangeDialog(PasswordChangeWrappedUI):
# Signal used when user close app after clicked esc or cros to close app.
# If user clicked Yes on message box return True other way return False.
closingSignal = pyqtSignal(QEvent)
callSignal = pyqtSignal(bool)
def __init__(self, client):
super(PasswordChangeDialog, self).__init__()
self.client = client
# self.__session_id = None
self.dialog = SelectAvatar(self)
self.loop = QEventLoop()
self.timer = QTimer()
self.timer.setSingleShot(True)
self.timer.timeout.connect(lambda: self.loop.exit(1))
self.set_change_password_button(self.change_password)
self.push_button_avatar.clicked.connect(self.selectAvatar)
def selectAvatar(self):
self.dialog.show()
def validatePasswords(self, password, repeat_password):
# TO DO
if len(password) < 8:
raise PasswdLengthError("Passwords is too short")
return False
if len(password) > 20:
raise PasswdLengthError("Passwords is too long")
return False
if password != repeat_password:
raise PasswdEqualError("Passwords are not equal")
return False
return True
def validateLogin(self, login):
# TO DO and CHECK
if login is None or login == "":
raise TypeError("Login is empty!")
return False
if type(login) != str:
raise TypeError("Login is not text!")
return False
return True
def validateData(self, login, password, repeat_password):
try:
if self.validateLogin(login):
if self.validatePasswords(password, repeat_password):
return True
except Exception as e:
# TO DO SHOW PASSWORD CHANGE STATUS
#self.showRegisterStatus(str(e.args[0]))
return False
def getActivationStatus(self):
print("[*] ActivationDialog info: Get response from server ",
self.client.received)
if self.client.received == "200 CHANGED":
status = "Status zmiany hasła | " + str(self.client.received)
self.showRegisterStatus(status)
return True
elif self.client.received == "406 NOT CHANGED":
status = "Status zmiany hasła | " + str(self.client.received)
self.showRegisterStatus(status)
return False
def change_password(self):
login = self.get_login()
avatar = self.get_avatar_name()
print("<*> Client info: avatar is:", avatar)
passwd = self.get_password_activate()
passwd_hash = hashlib.sha256(passwd.encode()).hexdigest()
repeat_passwd = self.get_repeat_password_activate()
if self.validateData(login, passwd, repeat_passwd):
payload = {
"type": "d",
"description": "CHANGE",
"NICKNAME": login,
"PASSWORD": passwd_hash,
"AVATAR": avatar
}
self.client.sendMessage(json.dumps(payload).encode("utf-8"))
return self.getActivationStatus()
else:
return False
def closeApp(self, event):
print(event)
self.close()
def closeEvent(self, event):
self.closingSignal.emit(event)
def setUserName(self, user_name):
self.username = user_name
def waiting_for_signal(self):
self.timer.start(10000) # 10 second time-out
print('{*} MainWindow info: waiting for response')
if self.loop.exec_() == 0:
self.timer.stop()
print("{*} MainWindow info: stop timer")
return True
else:
print('{!} MainWindow error: time-out :(')
return False
def keyPressEvent(self, event):
"""
Close application from escape key.
"""
if event.key() == Qt.Key_Escape:
self.close()
class CharacterDialog(WindowModalDialog):
def __init__(self, parent):
super(CharacterDialog, self).__init__(parent)
self.setWindowTitle(_("KeepKey Seed Recovery"))
self.character_pos = 0
self.word_pos = 0
self.loop = QEventLoop()
self.word_help = QLabel()
self.char_buttons = []
vbox = QVBoxLayout(self)
vbox.addWidget(WWLabel(CHARACTER_RECOVERY))
hbox = QHBoxLayout()
hbox.addWidget(self.word_help)
for i in range(4):
char_button = CharacterButton('*')
char_button.setMaximumWidth(36)
self.char_buttons.append(char_button)
hbox.addWidget(char_button)
self.accept_button = CharacterButton(_("Accept Word"))
self.accept_button.clicked.connect(partial(self.process_key, 32))
self.rejected.connect(partial(self.loop.exit, 1))
hbox.addWidget(self.accept_button)
hbox.addStretch(1)
vbox.addLayout(hbox)
self.finished_button = QPushButton(_("Seed Entered"))
self.cancel_button = QPushButton(_("Cancel"))
self.finished_button.clicked.connect(
partial(self.process_key, Qt.Key_Return))
self.cancel_button.clicked.connect(self.rejected)
buttons = Buttons(self.finished_button, self.cancel_button)
vbox.addSpacing(40)
vbox.addLayout(buttons)
self.refresh()
self.show()
def refresh(self):
self.word_help.setText("Enter seed word %2d:" % (self.word_pos + 1))
self.accept_button.setEnabled(self.character_pos >= 3)
self.finished_button.setEnabled((self.word_pos in (11, 17, 23)
and self.character_pos >= 3))
for n, button in enumerate(self.char_buttons):
button.setEnabled(n == self.character_pos)
if n == self.character_pos:
button.setFocus()
def is_valid_alpha_space(self, key):
# Auto-completion requires at least 3 characters
if key == ord(' ') and self.character_pos >= 3:
return True
# Firmware aborts protocol if the 5th character is non-space
if self.character_pos >= 4:
return False
return (key >= ord('a') and key <= ord('z')
or (key >= ord('A') and key <= ord('Z')))
def process_key(self, key):
self.data = None
if key == Qt.Key_Return and self.finished_button.isEnabled():
self.data = {'done': True}
elif key == Qt.Key_Backspace and (self.word_pos or self.character_pos):
self.data = {'delete': True}
elif self.is_valid_alpha_space(key):
self.data = {'character': chr(key).lower()}
if self.data:
self.loop.exit(0)
def keyPressEvent(self, event):
self.process_key(event.key())
if not self.data:
QDialog.keyPressEvent(self, event)
def get_char(self, word_pos, character_pos):
self.word_pos = word_pos
self.character_pos = character_pos
self.refresh()
if self.loop.exec_():
self.data = None # User cancelled
class EquityLinkedWarrant(QAxWidget):
def __init__(self):
super().__init__()
# 전역변수 모음
self.account_number = None
self.total_days = 0
self.stock_day_list = []
# self.target_code_list = ["032190","263750","091990","056190","032500","034950","225190","049520","225330","040420"]
self.target_code_list = []
self.market_type = None
self.stock_day_table = "stock_day_elw"
# 종목 일봉 데이터 수집 화면번호 지정
self.stock_day_screen_number = "4000"
# 짤려서 현재 실행할 순번 1001, 569
# 2020-10-05 23:27
# 2020-10-06 22:16 ~
# 2020-10-08 22:58 ~ 2020-10-09 00:30 완료
# 2020-10-12 MON
# 2020-10-13 TUE 20:01 ~
self.current_day = 0
# 수집 대상 기간 일수
self.days = 0
# 로그인 Event Loop 실행
self.login_event_loop = QEventLoop()
self.day_info_event_loop = QEventLoop()
# Kiwoom OCX 사용 함수 호출
self.get_ocx_instance()
# Event Slots
self.event_slots()
# 로그인 signal 함수 호출
self.signal_login_commConnect()
# 계좌정보 가져오기
self.get_login_info()
# 전체 종목 수집 및 분석을 위한 전체 종복 정보 가져오기
self.get_day_info_by_stock(sPrevNext="0")
def get_ocx_instance(self):
self.setControl("KHOPENAPI.KHOpenAPICtrl.1")
print(">> Kiwoom OCX KHOPENAPI.KHOpenAPICtrl.1 setControl")
def event_slots(self):
self.OnEventConnect.connect(self.login_event_slot)
self.OnReceiveTrData.connect(self.trdata_slot)
def signal_login_commConnect(self):
self.dynamicCall("CommConnect()")
self.login_event_loop.exec_()
def login_event_slot(self, error_code):
self.login_event_loop.exit()
def get_login_info(self):
accounts = self.dynamicCall("GetLoginInfo(QString)", "ACCNO")
self.account_number = accounts.split(";")[0]
print(">> 계좌번호: %s" % self.account_number)
def get_day_info_by_stock(self, sPrevNext="0"):
QTest.qWait(1000) # 3600 Delay Time
if (sPrevNext == '0'):
print(
"종목코드,종목명,발행사명,회차,기초자산명,권리구분,만기일,행사가격,현재가,이론가,대비구분,전일대비,등락율,거래량"
)
############################## opt30005 : ELW조건검색요청 ##############################
# 000000000000:전체, 3:한국투자증권, 5:미래대우, 6:신영, 12:NK투자, 17:KB증권
self.dynamicCall("SetInputValue(QString,QString)", "발행사코드",
"000000000000")
# 000000000000:전체, 201:KOSPI200, 150:KOSDAQ150, 005930:삼성전자, ...
self.dynamicCall("SetInputValue(QString,QString)", "기초자산코드",
"000000000000")
# 0:전체, 1:콜, 2:풋, 3:DC, 4:DP, 5:EX, 6:조기종료콜, 7:조기종료풋
self.dynamicCall("SetInputValue(QString,QString)", "권리구분", "0")
# 000000000000:전체, 3:한국투자증, 5:미래대우, 6:신영, 12:NK투자, 17:KB증권
self.dynamicCall("SetInputValue(QString,QString)", "LP코드",
"000000000000")
# 0:정렬없음, 1:상승율, 2:상승폭, 3:하락율, 4:하락폭, 5:거래량, 6:거래대금, 7:잔존일
self.dynamicCall("SetInputValue(QString,QString)", "정렬구분", "6")
self.dynamicCall("CommRqData(QString,QString,int,QString)",
"ELW조건검색요청", "opt30005", sPrevNext,
self.stock_day_screen_number)
self.day_info_event_loop.exec_()
def trdata_slot(self, screen_number, tr_name, tr_code, record_name,
sPrevNext):
# 멀티데이터 일봉 차트 가져오기
multi_cnt = self.dynamicCall("GetRepeatCnt(QString,QString)", tr_code,
tr_name)
for i in range(multi_cnt):
data = []
output01 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "종목코드").strip()
output02 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "발행사명").strip()
output03 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "회차").strip()
output04 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "기초자산명").strip()
output05 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "권리구분").strip()
output06 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "만기일").strip()
output07 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "현재가").strip()
output08 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "대비구분").strip()
output09 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "전일대비").strip()
output10 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "등락율").strip()
output11 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "거래량").strip()
output12 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "거래량대비").strip()
output13 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "거래대금").strip()
output14 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "전일거래량").strip()
output15 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "매도호가").strip()
output16 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "매수호가").strip()
output17 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "패리티").strip()
output18 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "기어링비율").strip()
output19 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "손일분기율").strip()
output20 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "자본지지점").strip()
output21 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "이론가").strip()
output22 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "내재변동성").strip()
output23 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "델타").strip()
output24 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "레버리지").strip()
output25 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "행사가격").strip()
output26 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "전환비율").strip()
output27 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "LP보유비율").strip()
output28 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "손익분기점").strip()
output29 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "최종거래일").strip()
output30 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "상장일").strip()
output31 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "LP초종공급일").strip()
output32 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "종목명").strip()
output33 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "잔존일수").strip()
output34 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "괴리율").strip()
output35 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "LP회원사명").strip()
output36 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "LP회원사명1").strip()
output37 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "LP회원사명2").strip()
output38 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "Xray순간체결량정리매매구분").strip()
output39 = self.dynamicCall(
"GetCommData(QString,QString,int,QString)", tr_code, tr_name,
i, "Xray순간체결량증거금100구분").strip()
self.total_days = self.total_days + 1
data.append("")
data.append(output01)
data.append(output02)
data.append(output03)
data.append(output04)
data.append(output05)
data.append(output06)
data.append(output07)
data.append(output08)
data.append(output09)
data.append(output10)
data.append(output11)
data.append(output12)
data.append(output13)
data.append(output14)
data.append(output15)
data.append(output16)
data.append(output17)
data.append(output18)
data.append(output19)
data.append(output20)
data.append(output21)
data.append(output22)
data.append(output23)
data.append(output24)
data.append(output25)
data.append(output26)
data.append(output27)
data.append(output28)
data.append(output29)
data.append(output30)
data.append(output31)
data.append(output32)
data.append(output33)
data.append(output34)
data.append(output35)
data.append(output36)
data.append(output37)
data.append(output38)
data.append(output39)
data.append("")
self.stock_day_list.append(data.copy())
output06 = output06[:6]
is_lp = False
if output02 == '한국투자증' or output02 == '미래대우':
is_lp = True
if output04 == 'KOSPI200' and is_lp and (
output06 == '202104' or output06 == '202105'
or output06 == '202106') and int(output11) > 0:
# if output04 == 'KOSPI200' and output02 == '한국투자증' and (output06 == '20210412' or output06 == '20210517' or output06 == '20210614') and int(output11) > 0:
# if output04 == 'KOSPI200' and (output02 == '한국투자증' or output02 == '미래대우') and (output06 == '20210118' or output06 == '20210216'):
# if output04 == 'KOSPI200' and int(output11) > 100000:
print("%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s" %
(output01, output32, output02, output03, output04,
output05, output06, output25, output07, output21,
output08, output09, output10, output11))
if self.total_days == self.days:
# print(">> %s,%s,%s,%s,%s,%s,%s,%s,%s " % ((i+1),stock_code,date,close_price,open_price,high_price,low_price,volume,trading_value))
sPrevNext = "0"
break
if sPrevNext == "2":
self.get_day_info_by_stock(sPrevNext=sPrevNext)
else:
self.day_info_event_loop.exit()
def insert_stock_day_list(self):
conn_string = "host='localhost' dbname='postgres' user='******' password='******' port='5432'"
conn = psycopg2.connect(conn_string)
cur = conn.cursor()
day_num = 0
for stock_day in self.stock_day_list:
delete_day_sql = "delete from " + self.stock_day_table + " where stock_code = %s and dt = %s"
cur.execute(delete_day_sql, (
stock_day[1],
stock_day[3],
))
day_num = day_num + 1
sql = "insert into " + self.stock_day_table + " (stock_code,stock_name,day_num,dt,volume,trading_value,close_price,open_price,high_price,low_price,market_type) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
cur.execute(
sql, (stock_day[1], stock_day[2], day_num, stock_day[3],
stock_day[4], stock_day[5], stock_day[6], stock_day[7],
stock_day[8], stock_day[9], stock_day[10]))
conn.commit()
cur.close()
conn.close()
def update_day_num(self):
conn_string = "host='localhost' dbname='postgres' user='******' password='******' port='5432'"
conn = psycopg2.connect(conn_string)
cur = conn.cursor()
# 시장유형의 모든 종목 입력 후 시장유형 별 일봉 순번 갱신
update_day_sql = " update " + self.stock_day_table + " s "
update_day_sql += " set day_num = a.day_num "
update_day_sql += " from "
update_day_sql += " ( "
update_day_sql += " select row_number() over(partition by stock_code order by dt desc) as day_num "
update_day_sql += " , stock_code "
update_day_sql += " , dt "
update_day_sql += " , market_type "
update_day_sql += " from " + self.stock_day_table + " "
update_day_sql += " where market_type = %s "
update_day_sql += " ) as a "
update_day_sql += " where s.stock_code = a.stock_code "
update_day_sql += " and s.dt = a.dt "
update_day_sql += " and s.market_type = a.market_type "
update_day_sql += " and s.market_type = %s "
cur.execute(update_day_sql, (
self.market_type,
self.market_type,
))
conn.commit()
cur.close()
conn.close()
class Browser(QObject):
comment_postdata_example = {
'ft_ent_identifier': '735272899920440', # ИД сообщения
'comment_text': 'Cool))', # Текст коментария
'source': '22',
'client_id': '1429632677205%3A3425397009',
'reply_fbid': '',
'parent_comment_id': '',
'rootid': 'u_ps_0_0_k',
'clp': '',
'attached_sticker_fbid': '0',
'attached_photo_fbid': '0',
'feed_context': '%7B%22fbfeed_context%22%3Atrue%2C%22location_type%22%3A36%2C%22is_starred%22%3Afalse%2C%22is_pinned_post%22%3Afalse%2C%22can_moderate_timeline_story%22%3Afalse%2C%22profile_id%22%3A308106089303792%2C%22outer_object_element_id%22%3A%22u_ps_0_0_0%22%2C%22object_element_id%22%3A%22u_ps_0_0_0%22%2C%22is_ad_preview%22%3Afalse%2C%22is_editable%22%3Afalse%7D',
'ft[tn]': '[]',
'ft[top_level_post_id]': '750869418360788',
'ft[fbfeed_location]': '36',
'nctr[_mod]': 'pagelet_timeline_main_column',
'av': '100009110845526',
'__user': '******',
'__a': '1',
'__dyn': '', #пустой
'__req': 'c',
'fb_dtsg': 'AQEkxiOYhtrJ', # инпут в теле документа
'ttstamp': '26581716611911872109105876676',
'__rev': '1713404',
}
def __init__(self, parent):
super().__init__(parent)
self.set_url('http://google.ru')
conn = QNetworkAccessManager()
self.conn = conn
self.r = QNetworkRequest()
self.r.attribute(QNetworkRequest.CookieSaveControlAttribute, QVariant(True))
# self.r.setHeader(QNetworkRequest.ContentTypeHeader, "application/x-www-form-urlencoded")
# self.r.setRawHeader("Referer", "http://www.facebook.com/")
# self.r.setRawHeader("Host", "www.facebook.com")
self.cj = QNetworkCookieJar()
conn.setCookieJar(self.cj)
conn.createRequest = self._create_request
self.wv = WebView()
self.wv.show()
self.wv.page().setNetworkAccessManager(conn)
# self.wv.auth()
self.loop = QEventLoop()
pass
def _create_request(self, operation, request, data):
# print(data)
reply = QNetworkAccessManager.createRequest(self.conn,
operation,
request,
data)
self.conn.new_reply = reply
self.wv_reply = reply
return reply
def set_url(self, url):
if isinstance(url, QByteArray):
self.url = QUrl().fromEncoded(url)
else:
self.url = QUrl(url)
def send_request(self, post=None, data={}):
loop = QEventLoop()
self.r.setUrl(self.url)
if post:
encoded_data = self._urlencode_post_data(data)
pprint(encoded_data)
self.reply_post = self.conn.post(self.r, encoded_data)
self.reply_post.downloadProgress.connect(self.prepare_responce)
else:
self.reply = self.conn.get(self.r)
self.reply.finished.connect(self.prepare_responce)
# return \
loop.exec()
def prepare_responce(self):
# self.check_redirect()
self.responce = self.reply_post.readAll()#.data().decode('utf-8')
pprint(self.responce)
sys.exit()
def check_redirect(self):
print(self.url)
a = self.reply.rawHeader('Location')
if len(a) > 0:
self.set_url(a)
self.send_request()
else:
self.loop.exit()
def test(self):
self.wv.auth('https://www.facebook.com/freelanceuidesignerdeveloper')
self.wv.authentication.connect(self.webview_login)
def _urlencode_post_data(self, post_data):
post_params = []
for (key, value) in post_data.items():
print(key, value)
post_params.append(key+'='+value)
return '&'.join(post_params)
def webview_login(self):
text_page = self.wv.page().mainFrame().toHtml()
data = {
'ft_ent_identifier': '735272899920440',
'comment_text': 'amazing',
'source': '22',
'client_id': '1429632677205%3A3425397009',
'reply_fbid': '',
'parent_comment_id': '',
'rootid': 'u_ps_0_0_o',
'clp': '',
'attached_sticker_fbid': '0',
'attached_photo_fbid': '0',
'feed_context': '%7B%22fbfeed_context%22%3Atrue%2C%22location_type%22%3A36%2C%22is_starred%22%3Afalse%2C%22is_pinned_post%22%3Afalse%2C%22can_moderate_timeline_story%22%3Afalse%2C%22profile_id%22%3A308106089303792%2C%22outer_object_element_id%22%3A%22u_ps_0_0_0%22%2C%22object_element_id%22%3A%22u_ps_0_0_0%22%2C%22is_ad_preview%22%3Afalse%2C%22is_editable%22%3Afalse%7D',
'ft[tn]': '[]',
'ft[top_level_post_id]': '',
'ft[fbfeed_location]': '36',
'nctr[_mod]': 'pagelet_timeline_main_column',
'av': '100009110845526',
'__user': '******',
'__a': '1',
'__dyn': '', #пустой
'__req': 'c',
'fb_dtsg': 'AQEkxiOYhtrJ', # инпут в теле документа
'ttstamp': '26581716611911872109105876676',
'__rev': '1713404',
}
dtsg_index = text_page.find('"token":"AQ')
data['fb_dtsg'] = text_page[dtsg_index+9:dtsg_index+21]
data['ttstamp'] = self.ttstamp_gen(data['fb_dtsg'])
data['ft_ent_identifier'] = '849713745045784' # self.get_post_id(post_object)
data['comment_text'] = random_number.choice(config.comments_list)
pprint(data['comment_text'])
# data['av'] = data['__user'] = from_user
# self.applyMetaData()
self.url = QUrl('https://www.facebook.com/ajax/ufi/add_comment.php')
self.send_request(True, data)
sys.exit()
def applyMetaData(self):
print('applyMetaData')
raw_header = {
'Host': 'www.facebook.com',
'User-Agent': ' Mozilla/5.0 (X11; Linux x86_64; rv:37.0) Gecko/20100101 Firefox/37.0',
'Accept': ' text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
'Accept-Language': ' en-US,en;q=0.5',
'Content-Type': ' application/x-www-form-urlencoded; charset=UTF-8',
'Referer': ' https://www.facebook.com/SoftProposal?fref=nf',
}
for (key, header) in raw_header.items():
self.r.setRawHeader(key, header)
@staticmethod
def ttstamp_gen(fb_dtsg):
u = ''
for v in fb_dtsg:
u += str(ord(v))
return '2'+u
@staticmethod
def get_post_id(post):
if post.object_id:
return post.object_id
else:
str = post.id.split('_')
if len(str) > 1:
return str[0]
else:
post.id
class ConverterThread(QThread):
"""Thread converts markdown to HTML.
"""
# This signal is emitted by the converter thread when a file has been
# converted to HTML.
htmlReady = pyqtSignal(
# Path to the file which should be converted to / displayed as HTML.
str,
# HTML rendering of the file; empty if the HTML is provided in a file
# specified by the URL below.
str,
# Error text resulting from the conversion process.
str,
# A reference to a file containing HTML rendering. Empty if the second
# parameter above contains the HTML instead.
QUrl)
# This signal clears the context of the log window.
logWindowClear = pyqtSignal()
# This signal emits messages for the log window.
logWindowText = pyqtSignal(
# A string to append to the log window.
str)
_Task = collections.namedtuple("Task", ["filePath", "language", "text"])
def __init__(self):
QThread.__init__(self)
self._queue = queue.Queue()
self.start(QThread.LowPriority)
self._ac = AsyncController('QThread', self)
self._ac.defaultPriority = QThread.LowPriority
self._SphinxInvocationCount = 1
def process(self, filePath, language, text):
"""Convert data and emit result.
"""
self._queue.put(self._Task(filePath, language, text))
def stop_async(self):
self._queue.put(None)
def _getHtml(self, language, text, filePath):
"""Get HTML for document
"""
if language == 'Markdown':
return self._convertMarkdown(text), None, QUrl()
# For ReST, use docutils only if Sphinx isn't available.
elif language == 'Restructured Text' and not sphinxEnabledForFile(filePath):
htmlUnicode, errString = self._convertReST(text)
return htmlUnicode, errString, QUrl()
elif filePath and sphinxEnabledForFile(filePath): # Use Sphinx to generate the HTML if possible.
return self._convertSphinx(filePath)
elif filePath and canUseCodeChat(filePath): # Otherwise, fall back to using CodeChat+docutils.
return self._convertCodeChat(text, filePath)
else:
return 'No preview for this type of file', None, QUrl()
def _convertMarkdown(self, text):
"""Convert Markdown to HTML
"""
try:
import markdown
except ImportError:
return 'Markdown preview requires <i>python-markdown</i> package<br/>' \
'Install it with your package manager or see ' \
'<a href="http://packages.python.org/Markdown/install.html">installation instructions</a>'
extensions = ['fenced_code', 'nl2br', 'tables', 'enki.plugins.preview.mdx_math']
# version 2.0 supports only extension names, not instances
if markdown.version_info[0] > 2 or \
(markdown.version_info[0] == 2 and markdown.version_info[1] > 0):
class _StrikeThroughExtension(markdown.Extension):
"""http://achinghead.com/python-markdown-adding-insert-delete.html
Class is placed here, because depends on imported markdown, and markdown import is lazy
"""
DEL_RE = r'(~~)(.*?)~~'
def extendMarkdown(self, md, md_globals):
# Create the del pattern
delTag = markdown.inlinepatterns.SimpleTagPattern(self.DEL_RE, 'del')
# Insert del pattern into markdown parser
md.inlinePatterns.add('del', delTag, '>not_strong')
extensions.append(_StrikeThroughExtension())
return markdown.markdown(text, extensions)
def _convertReST(self, text):
"""Convert ReST
"""
try:
import docutils.core
import docutils.writers.html4css1
except ImportError:
return 'Restructured Text preview requires the <i>python-docutils</i> package.<br/>' \
'Install it with your package manager or see ' \
'<a href="http://pypi.python.org/pypi/docutils"/>this page.</a>', None
errStream = io.StringIO()
settingsDict = {
# Make sure to use Unicode everywhere.
'output_encoding': 'unicode',
'input_encoding' : 'unicode',
# Don't stop processing, no matter what.
'halt_level' : 5,
# Capture errors to a string and return it.
'warning_stream' : errStream }
# Frozen-specific settings.
if isFrozen:
settingsDict['template'] = (
# The default docutils stylesheet and template uses a relative path,
# which doesn't work when frozen ???. Under Unix when not frozen,
# it produces:
# ``IOError: [Errno 2] No such file or directory:
# '/usr/lib/python2.7/dist-packages/docutils/writers/html4css1/template.txt'``.
os.path.join(os.path.dirname(docutils.writers.html4css1.__file__),
docutils.writers.html4css1.Writer.default_template) )
settingsDict['stylesheet_dirs'] = ['.',
os.path.dirname(docutils.writers.html4css1.__file__)]
htmlString = docutils.core.publish_string(text, writer_name='html',
settings_overrides=settingsDict)
errString = errStream.getvalue()
errStream.close()
return htmlString, errString
def _convertSphinx(self, filePath):
# Run the builder.
errString = self._runHtmlBuilder()
# Look for the HTML output.
#
# Get an absolute path to the output path, which could be relative.
outputPath = core.config()['Sphinx']['OutputPath']
projectPath = core.config()['Sphinx']['ProjectPath']
if not os.path.isabs(outputPath):
outputPath = os.path.join(projectPath, outputPath)
# Create an htmlPath as OutputPath + remainder of filePath.
htmlPath = os.path.join(outputPath + filePath[len(projectPath):])
html_file_suffix = '.html'
try:
with codecs.open(os.path.join(projectPath, 'sphinx-enki-info.txt')) as f:
hfs = f.read()
# If the file is empty, then html_file_suffix wasn't defined
# or is None. In this case, use the default extension.
# Otherwise, use the extension read from the file.
if hfs:
html_file_suffix = hfs
except:
errString = "Warning: assuming .html extension. Use " + \
"the conf.py template to set the extension.\n" + errString
pass
# First place to look: file.html. For example, look for foo.py
# in foo.py.html.
htmlFile = htmlPath + html_file_suffix
# Second place to look: file without extension.html. For
# example, look for foo.html for foo.rst.
htmlFileAlter = os.path.splitext(htmlPath)[0] + html_file_suffix
# Check that the output file produced by Sphinx is newer than
# the source file it was built from.
if os.path.exists(htmlFile):
return _checkModificationTime(filePath, htmlFile, errString)
elif os.path.exists(htmlFileAlter):
return _checkModificationTime(filePath, htmlFileAlter, errString)
else:
return ('No preview for this type of file.<br>Expected ' +
htmlFile + " or " + htmlFileAlter, errString, QUrl())
def _convertCodeChat(self, text, filePath):
# Use StringIO to pass CodeChat compilation information back to
# the UI.
errStream = io.StringIO()
try:
htmlString = CodeToRest.code_to_html_string(text, errStream,
filename=filePath)
except KeyError:
# Although the file extension may be in the list of supported
# extensions, CodeChat may not support the lexer chosen by Pygments.
# For example, a ``.v`` file may be Verilog (supported by CodeChat)
# or Coq (not supported). In this case, provide an error messsage
errStream.write('Error: this file is not supported by CodeChat.')
htmlString = ''
errString = errStream.getvalue()
errStream.close()
return htmlString, errString, QUrl()
def _runHtmlBuilder(self):
# Build the commond line for Sphinx.
if core.config()['Sphinx']['AdvancedMode']:
htmlBuilderCommandLine = core.config()['Sphinx']['Cmdline']
if sys.platform.startswith('linux'):
# If Linux is used, then subprocess cannot take the whole
# commandline as the name of an executable file. Module shlex
# has to be used to parse commandline.
htmlBuilderCommandLine = shlex.split(htmlBuilderCommandLine)
else:
# For available builder options, refer to: http://sphinx-doc.org/builders.html
htmlBuilderCommandLine = [core.config()['Sphinx']['Executable'],
# Place doctrees in the ``_build`` directory; by default, Sphinx
# places this in _build/html/.doctrees.
'-d', os.path.join('_build', 'doctrees'),
# Source directory -- the current directory, since we'll chdir to
# the project directory before executing this.
'.',
# Build directory
core.config()['Sphinx']['OutputPath']]
# Invoke it.
try:
# Clear the log at the beginning of a Sphinx build.
self.logWindowClear.emit()
cwd = core.config()['Sphinx']['ProjectPath']
# If the command line is already a string (advanced mode), just print it.
# Otherwise, it's a list that should be transformed to a string.
if isinstance(htmlBuilderCommandLine, str):
htmlBuilderCommandLineStr = htmlBuilderCommandLine
else:
htmlBuilderCommandLineStr = ' '.join(htmlBuilderCommandLine)
self.logWindowText.emit('{} : {}\n\n'.format(cwd,
htmlBuilderCommandLineStr))
# Run Sphinx, reading stdout in a separate thread.
self._qe = QEventLoop()
# Sphinx will output just a carriage return (0x0D) to simulate a
# single line being updated by build status and the build
# progresses. Without universal newline support here, we'll wait
# until the build is complete (with a \n\r) to report any build
# progress! So, enable universal newlines, so that each \r will be
# treated as a separate line, providing immediate feedback on build
# progress.
popen = open_console_output(htmlBuilderCommandLine, cwd=cwd,
universal_newlines=True)
# Perform reads in an event loop. The loop is exit when all reads
# have completed. We can't simply start the _stderr_read thread
# here, because calls to self._qe_exit() will be ignored until
# we're inside the event loop.
QTimer.singleShot(0, lambda: self._popen_read(popen))
self._qe.exec_()
except OSError as ex:
return (
'Failed to execute HTML builder:\n'
'{}\n'.format(str(ex)) +
'Go to Settings -> Settings -> CodeChat to set HTML'
' builder configurations.')
return self._stderr
# Read from stdout (in this thread) and stderr (in another thread),
# so that the user sees output as the build progresses, rather than only
# producing output after the build is complete.
def _popen_read(self, popen):
# Read are blocking; we can't read from both stdout and stderr in the
# same thread without possible buffer overflows. So, use this thread to
# read from and immediately report progress from stdout. In another
# thread, read all stderr and report that after the build finishes.
self._ac.start(None, self._stderr_read, popen.stderr)
# Read a line of stdout then report it to the user immediately.
s = popen.stdout.readline()
while s:
self.logWindowText.emit(s.rstrip('\n'))
s = popen.stdout.readline()
self._SphinxInvocationCount += 1
# I would expect the following code to do the same thing. It doesn't:
# instead, it waits until Sphinx completes before returning anything.
# ???
#
# .. code-block: python
# :linenos:
#
# for s in popen.stdout:
# self.logWindowText.emit(s)
# Runs in a separate thread to read stdout. It then exits the QEventLoop as
# a way to signal that stderr reads have completed.
def _stderr_read(self, stderr):
self._stderr = stderr.read()
self._qe.exit()
def run(self):
"""Thread function
"""
while True: # exits with break
# wait task
task = self._queue.get()
# take the last task
while self._queue.qsize():
task = self._queue.get()
if task is None: # None is a quit command
self._ac.terminate()
break
# TODO: This is ugly. Should pass this exception back to the main
# thread and re-raise it there, or use a QFuture like approach which
# does this automaticlaly.
try:
html, errString, url = self._getHtml(task.language, task.text,
task.filePath)
except Exception:
traceback.print_exc()
self.htmlReady.emit(task.filePath, html, errString, url)
# Free resources.
self._ac.terminate()
class CategoryPopupMenu(FramelessWindow):
triggered = Signal(QAction)
hovered = Signal(QAction)
def __init__(self, parent=None, **kwargs):
FramelessWindow.__init__(self, parent, **kwargs)
self.setWindowFlags(self.windowFlags() | Qt.Popup)
layout = QVBoxLayout()
layout.setContentsMargins(6, 6, 6, 6)
self.__menu = MenuPage()
self.__menu.setActionRole(QtWidgetRegistry.WIDGET_ACTION_ROLE)
if sys.platform == "darwin":
self.__menu.view().setAttribute(Qt.WA_MacShowFocusRect, False)
self.__menu.triggered.connect(self.__onTriggered)
self.__menu.hovered.connect(self.hovered)
self.__dragListener = ItemViewDragStartEventListener(self)
self.__dragListener.dragStarted.connect(self.__onDragStarted)
self.__menu.view().viewport().installEventFilter(self.__dragListener)
layout.addWidget(self.__menu)
self.setLayout(layout)
self.__action = None
self.__loop = None
self.__item = None
def setCategoryItem(self, item):
"""
Set the category root item (:class:`QStandardItem`).
"""
self.__item = item
model = item.model()
self.__menu.setModel(model)
self.__menu.setRootIndex(item.index())
def popup(self, pos=None):
if pos is None:
pos = self.pos()
geom = widget_popup_geometry(pos, self)
self.setGeometry(geom)
self.show()
def exec_(self, pos=None):
self.popup(pos)
self.__loop = QEventLoop()
self.__action = None
self.__loop.exec_()
self.__loop = None
if self.__action is not None:
action = self.__action
else:
action = None
return action
def hideEvent(self, event):
if self.__loop is not None:
self.__loop.exit(0)
return FramelessWindow.hideEvent(self, event)
def __onTriggered(self, action):
self.__action = action
self.triggered.emit(action)
self.hide()
if self.__loop:
self.__loop.exit(0)
def __onDragStarted(self, index):
desc = qunwrap(index.data(QtWidgetRegistry.WIDGET_DESC_ROLE))
icon = qunwrap(index.data(Qt.DecorationRole))
drag_data = QMimeData()
drag_data.setData(
"application/vnv.orange-canvas.registry.qualified-name",
desc.qualified_name)
drag = QDrag(self)
drag.setPixmap(icon.pixmap(38))
drag.setMimeData(drag_data)
# TODO: Should animate (accept) hide.
self.hide()
# When a drag is started and the menu hidden the item's tool tip
# can still show for a short time UNDER the cursor preventing a
# drop.
viewport = self.__menu.view().viewport()
filter = ToolTipEventFilter()
viewport.installEventFilter(filter)
drag.exec_(Qt.CopyAction)
viewport.removeEventFilter(filter)
class InstallWizard(QDialog, MessageBoxMixin, BaseWizard):
accept_signal = pyqtSignal()
def __init__(self, config, app, plugins):
QDialog.__init__(self, None)
BaseWizard.__init__(self, config, plugins)
self.setWindowTitle('Vialectrum - ' + _('Install Wizard'))
self.app = app
self.config = config
# Set for base base class
self.language_for_seed = config.get('language')
self.setMinimumSize(600, 400)
self.accept_signal.connect(self.accept)
self.title = QLabel()
self.main_widget = QWidget()
self.back_button = QPushButton(_("Back"), self)
self.back_button.setText(_('Back') if self.can_go_back() else _('Cancel'))
self.next_button = QPushButton(_("Next"), self)
self.next_button.setDefault(True)
self.logo = QLabel()
self.please_wait = QLabel(_("Please wait..."))
self.please_wait.setAlignment(Qt.AlignCenter)
self.icon_filename = None
self.loop = QEventLoop()
self.rejected.connect(lambda: self.loop.exit(0))
self.back_button.clicked.connect(lambda: self.loop.exit(1))
self.next_button.clicked.connect(lambda: self.loop.exit(2))
outer_vbox = QVBoxLayout(self)
inner_vbox = QVBoxLayout()
inner_vbox.addWidget(self.title)
inner_vbox.addWidget(self.main_widget)
inner_vbox.addStretch(1)
inner_vbox.addWidget(self.please_wait)
inner_vbox.addStretch(1)
scroll_widget = QWidget()
scroll_widget.setLayout(inner_vbox)
scroll = QScrollArea()
scroll.setWidget(scroll_widget)
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
scroll.setWidgetResizable(True)
icon_vbox = QVBoxLayout()
icon_vbox.addWidget(self.logo)
icon_vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addLayout(icon_vbox)
hbox.addSpacing(5)
hbox.addWidget(scroll)
hbox.setStretchFactor(scroll, 1)
outer_vbox.addLayout(hbox)
outer_vbox.addLayout(Buttons(self.back_button, self.next_button))
self.set_icon('vialectrum.png')
self.show()
self.raise_()
self.refresh_gui() # Need for QT on MacOSX. Lame.
def select_storage(self, path, get_wallet_from_daemon) -> Tuple[str, Optional[WalletStorage]]:
vbox = QVBoxLayout()
hbox = QHBoxLayout()
hbox.addWidget(QLabel(_('Wallet') + ':'))
self.name_e = QLineEdit()
hbox.addWidget(self.name_e)
button = QPushButton(_('Choose...'))
hbox.addWidget(button)
vbox.addLayout(hbox)
self.msg_label = QLabel('')
vbox.addWidget(self.msg_label)
hbox2 = QHBoxLayout()
self.pw_e = QLineEdit('', self)
self.pw_e.setFixedWidth(150)
self.pw_e.setEchoMode(2)
self.pw_label = QLabel(_('Password') + ':')
hbox2.addWidget(self.pw_label)
hbox2.addWidget(self.pw_e)
hbox2.addStretch()
vbox.addLayout(hbox2)
self.set_layout(vbox, title=_('Vialectrum wallet'))
self.temp_storage = WalletStorage(path, manual_upgrades=True)
wallet_folder = os.path.dirname(self.temp_storage.path)
def on_choose():
path, __ = QFileDialog.getOpenFileName(self, "Select your wallet file", wallet_folder)
if path:
self.name_e.setText(path)
def on_filename(filename):
path = os.path.join(wallet_folder, filename)
wallet_from_memory = get_wallet_from_daemon(path)
try:
if wallet_from_memory:
self.temp_storage = wallet_from_memory.storage
else:
self.temp_storage = WalletStorage(path, manual_upgrades=True)
self.next_button.setEnabled(True)
except BaseException:
self.logger.exception('')
self.temp_storage = None
self.next_button.setEnabled(False)
user_needs_to_enter_password = False
if self.temp_storage:
if not self.temp_storage.file_exists():
msg =_("This file does not exist.") + '\n' \
+ _("Press 'Next' to create this wallet, or choose another file.")
elif not wallet_from_memory:
if self.temp_storage.is_encrypted_with_user_pw():
msg = _("This file is encrypted with a password.") + '\n' \
+ _('Enter your password or choose another file.')
user_needs_to_enter_password = True
elif self.temp_storage.is_encrypted_with_hw_device():
msg = _("This file is encrypted using a hardware device.") + '\n' \
+ _("Press 'Next' to choose device to decrypt.")
else:
msg = _("Press 'Next' to open this wallet.")
else:
msg = _("This file is already open in memory.") + "\n" \
+ _("Press 'Next' to create/focus window.")
else:
msg = _('Cannot read file')
self.msg_label.setText(msg)
if user_needs_to_enter_password:
self.pw_label.show()
self.pw_e.show()
self.pw_e.setFocus()
else:
self.pw_label.hide()
self.pw_e.hide()
button.clicked.connect(on_choose)
self.name_e.textChanged.connect(on_filename)
n = os.path.basename(self.temp_storage.path)
self.name_e.setText(n)
while True:
if self.loop.exec_() != 2: # 2 = next
raise UserCancelled
if self.temp_storage.file_exists() and not self.temp_storage.is_encrypted():
break
if not self.temp_storage.file_exists():
break
wallet_from_memory = get_wallet_from_daemon(self.temp_storage.path)
if wallet_from_memory:
raise WalletAlreadyOpenInMemory(wallet_from_memory)
if self.temp_storage.file_exists() and self.temp_storage.is_encrypted():
if self.temp_storage.is_encrypted_with_user_pw():
password = self.pw_e.text()
try:
self.temp_storage.decrypt(password)
break
except InvalidPassword as e:
self.show_message(title=_('Error'), msg=str(e))
continue
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=str(e))
raise UserCancelled()
elif self.temp_storage.is_encrypted_with_hw_device():
try:
self.run('choose_hw_device', HWD_SETUP_DECRYPT_WALLET, storage=self.temp_storage)
except InvalidPassword as e:
self.show_message(title=_('Error'),
msg=_('Failed to decrypt using this hardware device.') + '\n' +
_('If you use a passphrase, make sure it is correct.'))
self.reset_stack()
return self.select_storage(path, get_wallet_from_daemon)
except BaseException as e:
self.logger.exception('')
self.show_message(title=_('Error'), msg=str(e))
raise UserCancelled()
if self.temp_storage.is_past_initial_decryption():
break
else:
raise UserCancelled()
else:
raise Exception('Unexpected encryption version')
return self.temp_storage.path, (self.temp_storage if self.temp_storage.file_exists() else None) #
def run_upgrades(self, storage):
path = storage.path
if storage.requires_split():
self.hide()
msg = _("The wallet '{}' contains multiple accounts, which are no longer supported since Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?").format(path)
if not self.question(msg):
return
file_list = '\n'.join(storage.split_accounts())
msg = _('Your accounts have been moved to') + ':\n' + file_list + '\n\n'+ _('Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
# raise now, to avoid having the old storage opened
raise UserCancelled()
action = storage.get_action()
if action and storage.requires_upgrade():
raise WalletFileException('Incomplete wallet files cannot be upgraded.')
if action:
self.hide()
msg = _("The file '{}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path)
if not self.question(msg):
if self.question(_("Do you want to delete '{}'?").format(path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
self.data = storage.db.data # FIXME
self.run(action)
for k, v in self.data.items():
storage.put(k, v)
storage.write()
return
if storage.requires_upgrade():
self.upgrade_storage(storage)
def finished(self):
"""Called in hardware client wrapper, in order to close popups."""
return
def on_error(self, exc_info):
if not isinstance(exc_info[1], UserCancelled):
self.logger.error("on_error", exc_info=exc_info)
self.show_error(str(exc_info[1]))
def set_icon(self, filename):
prior_filename, self.icon_filename = self.icon_filename, filename
self.logo.setPixmap(QPixmap(icon_path(filename))
.scaledToWidth(60, mode=Qt.SmoothTransformation))
return prior_filename
def set_layout(self, layout, title=None, next_enabled=True):
self.title.setText("<b>%s</b>"%title if title else "")
self.title.setVisible(bool(title))
# Get rid of any prior layout by assigning it to a temporary widget
prior_layout = self.main_widget.layout()
if prior_layout:
QWidget().setLayout(prior_layout)
self.main_widget.setLayout(layout)
self.back_button.setEnabled(True)
self.next_button.setEnabled(next_enabled)
if next_enabled:
self.next_button.setFocus()
self.main_widget.setVisible(True)
self.please_wait.setVisible(False)
def exec_layout(self, layout, title=None, raise_on_cancel=True,
next_enabled=True):
self.set_layout(layout, title, next_enabled)
result = self.loop.exec_()
if not result and raise_on_cancel:
raise UserCancelled
if result == 1:
raise GoBack from None
self.title.setVisible(False)
self.back_button.setEnabled(False)
self.next_button.setEnabled(False)
self.main_widget.setVisible(False)
self.please_wait.setVisible(True)
self.refresh_gui()
return result
def refresh_gui(self):
# For some reason, to refresh the GUI this needs to be called twice
self.app.processEvents()
self.app.processEvents()
def remove_from_recently_open(self, filename):
self.config.remove_from_recently_open(filename)
def text_input(self, title, message, is_valid, allow_multi=False):
slayout = KeysLayout(parent=self, header_layout=message, is_valid=is_valid,
allow_multi=allow_multi)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_text()
def seed_input(self, title, message, is_seed, options):
slayout = SeedLayout(title=message, is_seed=is_seed, options=options, parent=self)
self.exec_layout(slayout, title, next_enabled=False)
return slayout.get_seed(), slayout.is_bip39, slayout.is_ext
@wizard_dialog
def add_xpub_dialog(self, title, message, is_valid, run_next, allow_multi=False, show_wif_help=False):
header_layout = QHBoxLayout()
label = WWLabel(message)
label.setMinimumWidth(400)
header_layout.addWidget(label)
if show_wif_help:
header_layout.addWidget(InfoButton(WIF_HELP_TEXT), alignment=Qt.AlignRight)
return self.text_input(title, header_layout, is_valid, allow_multi)
@wizard_dialog
def add_cosigner_dialog(self, run_next, index, is_valid):
title = _("Add Cosigner") + " %d"%index
message = ' '.join([
_('Please enter the master public key (xpub) of your cosigner.'),
_('Enter their master private key (xprv) if you want to be able to sign for them.')
])
return self.text_input(title, message, is_valid)
@wizard_dialog
def restore_seed_dialog(self, run_next, test):
options = []
if self.opt_ext:
options.append('ext')
if self.opt_bip39:
options.append('bip39')
title = _('Enter Seed')
message = _('Please enter your seed phrase in order to restore your wallet.')
return self.seed_input(title, message, test, options)
@wizard_dialog
def confirm_seed_dialog(self, run_next, test):
self.app.clipboard().clear()
title = _('Confirm Seed')
message = ' '.join([
_('Your seed is important!'),
_('If you lose your seed, your money will be permanently lost.'),
_('To make sure that you have properly saved your seed, please retype it here.')
])
seed, is_bip39, is_ext = self.seed_input(title, message, test, None)
return seed
@wizard_dialog
def show_seed_dialog(self, run_next, seed_text):
title = _("Your wallet generation seed is:")
slayout = SeedLayout(seed=seed_text, title=title, msg=True, options=['ext'])
self.exec_layout(slayout)
return slayout.is_ext
def pw_layout(self, msg, kind, force_disable_encrypt_cb):
playout = PasswordLayout(msg=msg, kind=kind, OK_button=self.next_button,
force_disable_encrypt_cb=force_disable_encrypt_cb)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.new_password(), playout.encrypt_cb.isChecked()
@wizard_dialog
def request_password(self, run_next, force_disable_encrypt_cb=False):
"""Request the user enter a new password and confirm it. Return
the password or None for no password."""
return self.pw_layout(MSG_ENTER_PASSWORD, PW_NEW, force_disable_encrypt_cb)
@wizard_dialog
def request_storage_encryption(self, run_next):
playout = PasswordLayoutForHW(MSG_HW_STORAGE_ENCRYPTION)
playout.encrypt_cb.setChecked(True)
self.exec_layout(playout.layout())
return playout.encrypt_cb.isChecked()
@wizard_dialog
def confirm_dialog(self, title, message, run_next):
self.confirm(message, title)
def confirm(self, message, title):
label = WWLabel(message)
vbox = QVBoxLayout()
vbox.addWidget(label)
self.exec_layout(vbox, title)
@wizard_dialog
def action_dialog(self, action, run_next):
self.run(action)
def terminate(self, **kwargs):
self.accept_signal.emit()
def waiting_dialog(self, task, msg, on_finished=None):
label = WWLabel(msg)
vbox = QVBoxLayout()
vbox.addSpacing(100)
label.setMinimumWidth(300)
label.setAlignment(Qt.AlignCenter)
vbox.addWidget(label)
self.set_layout(vbox, next_enabled=False)
self.back_button.setEnabled(False)
t = threading.Thread(target=task)
t.start()
while True:
t.join(1.0/60)
if t.is_alive():
self.refresh_gui()
else:
break
if on_finished:
on_finished()
@wizard_dialog
def choice_dialog(self, title, message, choices, run_next):
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
clayout = ChoicesLayout(message, c_titles)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, title)
action = c_values[clayout.selected_index()]
return action
def query_choice(self, msg, choices):
"""called by hardware wallets"""
clayout = ChoicesLayout(msg, choices)
vbox = QVBoxLayout()
vbox.addLayout(clayout.layout())
self.exec_layout(vbox, '')
return clayout.selected_index()
@wizard_dialog
def choice_and_line_dialog(self, title: str, message1: str, choices: List[Tuple[str, str, str]],
message2: str, test_text: Callable[[str], int],
run_next, default_choice_idx: int=0) -> Tuple[str, str]:
vbox = QVBoxLayout()
c_values = [x[0] for x in choices]
c_titles = [x[1] for x in choices]
c_default_text = [x[2] for x in choices]
def on_choice_click(clayout):
idx = clayout.selected_index()
line.setText(c_default_text[idx])
clayout = ChoicesLayout(message1, c_titles, on_choice_click,
checked_index=default_choice_idx)
vbox.addLayout(clayout.layout())
vbox.addSpacing(50)
vbox.addWidget(WWLabel(message2))
line = QLineEdit()
def on_text_change(text):
self.next_button.setEnabled(test_text(text))
line.textEdited.connect(on_text_change)
on_choice_click(clayout) # set default text for "line"
vbox.addWidget(line)
self.exec_layout(vbox, title)
choice = c_values[clayout.selected_index()]
return str(line.text()), choice
@wizard_dialog
def line_dialog(self, run_next, title, message, default, test, warning='',
presets=(), warn_issue4566=False):
vbox = QVBoxLayout()
vbox.addWidget(WWLabel(message))
line = QLineEdit()
line.setText(default)
def f(text):
self.next_button.setEnabled(test(text))
if warn_issue4566:
text_whitespace_normalised = ' '.join(text.split())
warn_issue4566_label.setVisible(text != text_whitespace_normalised)
line.textEdited.connect(f)
vbox.addWidget(line)
vbox.addWidget(WWLabel(warning))
warn_issue4566_label = WWLabel(MSG_PASSPHRASE_WARN_ISSUE4566)
warn_issue4566_label.setVisible(False)
vbox.addWidget(warn_issue4566_label)
for preset in presets:
button = QPushButton(preset[0])
button.clicked.connect(lambda __, text=preset[1]: line.setText(text))
button.setMinimumWidth(150)
hbox = QHBoxLayout()
hbox.addWidget(button, alignment=Qt.AlignCenter)
vbox.addLayout(hbox)
self.exec_layout(vbox, title, next_enabled=test(default))
return line.text()
@wizard_dialog
def show_xpub_dialog(self, xpub, run_next):
msg = ' '.join([
_("Here is your master public key."),
_("Please share it with your cosigners.")
])
vbox = QVBoxLayout()
layout = SeedLayout(xpub, title=msg, icon=False, for_seed_words=False)
vbox.addLayout(layout.layout())
self.exec_layout(vbox, _('Master Public Key'))
return None
def init_network(self, network):
message = _("Electrum communicates with remote servers to get "
"information about your transactions and addresses. The "
"servers all fulfill the same purpose only differing in "
"hardware. In most cases you simply want to let Electrum "
"pick one at random. However if you prefer feel free to "
"select a server manually.")
choices = [_("Auto connect"), _("Select server manually")]
title = _("How do you want to connect to a server? ")
clayout = ChoicesLayout(message, choices)
self.back_button.setText(_('Cancel'))
self.exec_layout(clayout.layout(), title)
r = clayout.selected_index()
if r == 1:
nlayout = NetworkChoiceLayout(network, self.config, wizard=True)
if self.exec_layout(nlayout.layout()):
nlayout.accept()
else:
network.auto_connect = True
self.config.set_key('auto_connect', True, True)
@wizard_dialog
def multisig_dialog(self, run_next):
cw = CosignWidget(2, 2)
m_edit = QSlider(Qt.Horizontal, self)
n_edit = QSlider(Qt.Horizontal, self)
n_edit.setMinimum(2)
n_edit.setMaximum(15)
m_edit.setMinimum(1)
m_edit.setMaximum(2)
n_edit.setValue(2)
m_edit.setValue(2)
n_label = QLabel()
m_label = QLabel()
grid = QGridLayout()
grid.addWidget(n_label, 0, 0)
grid.addWidget(n_edit, 0, 1)
grid.addWidget(m_label, 1, 0)
grid.addWidget(m_edit, 1, 1)
def on_m(m):
m_label.setText(_('Require {0} signatures').format(m))
cw.set_m(m)
def on_n(n):
n_label.setText(_('From {0} cosigners').format(n))
cw.set_n(n)
m_edit.setMaximum(n)
n_edit.valueChanged.connect(on_n)
m_edit.valueChanged.connect(on_m)
on_n(2)
on_m(2)
vbox = QVBoxLayout()
vbox.addWidget(cw)
vbox.addWidget(WWLabel(_("Choose the number of signatures needed to unlock funds in your wallet:")))
vbox.addLayout(grid)
self.exec_layout(vbox, _("Multi-Signature Wallet"))
m = int(m_edit.value())
n = int(n_edit.value())
return (m, n)