# -*- coding: utf-8 -*- import os, sys, time from PyQt5.QtCore import Qt, QEvent, QThread, pyqtSignal, QRect, QRectF, QTimer, QPoint from PyQt5.QtWidgets import QApplication, QWidget, QDesktopWidget, QLabel from PyQt5.QtGui import QIcon, QPixmap, QPainter if sys.platform == "win32": pipe_path = '\\\\.\\pipe\\oneshot-journal-to-game' import ctypes.wintypes buff = ctypes.create_unicode_buffer(ctypes.wintypes.MAX_PATH) ctypes.windll.shell32.SHGetFolderPathW(None, CSIDL_PERSONAL, None, SHGFP_TYPE_CURRENT, buf) documents_path = os.path.join(buf.value, 'My Games') else: pipe_path = '/tmp/oneshot-pipe' documents_path = os.path.expanduser('~/Documents') left_close = False if sys.platform == "darwin": left_close = True try: base_path = sys._MEIPASS except AttributeError: base_path = os.path.abspath('.') class WatchPipe(QThread): change_image = pyqtSignal(str) def run(self): while True: while True: if os.path.exists(pipe_path): break else: time.sleep(0.1) pipe = open(pipe_path, 'r') pipe.flush() while True: if not os.path.exists(pipe_path): break # Handle game quitting cleanup message = os.read(pipe.fileno(), 256) if len(message) > 0: self.change_image.emit(message.decode()) time.sleep(0.05) class AnimationTimer(QThread): next_frame = pyqtSignal() def run(self): while True: self.next_frame.emit() time.sleep(1.0 / 60) class Journal(QWidget): def __init__(self, *args, **kwargs): self.app = kwargs['app'] del kwargs['app'] super().__init__(*args, **kwargs) self.mousedown = False self.mousedownpos = QPoint(0, 0) self.label = QLabel(self) self.change_image('default_en') self.close_label = QLabel(self) self.close_label.setPixmap(QPixmap(os.path.join(base_path, 'images', 'close.png'))) if not left_close: self.close_label.move(776, 0) # X = 800-24 if "linux" in sys.platform: self.setWindowFlags(Qt.FramelessWindowHint) else: self.setWindowFlags(self.windowFlags() | Qt.FramelessWindowHint | Qt.NoDropShadowWindowHint) self.setAttribute(Qt.WA_TranslucentBackground) self.setMouseTracking(True) self.setWindowTitle(' ') self.setMinimumSize(800, 600) self.setMaximumSize(800, 600) self.setGeometry(0, 0, 800, 600) self.show() def mousePressEvent(self, event): self.mousedown = True self.mousedownpos = event.pos() if ((left_close and self.mousedownpos.x() <= 24) or (not left_close and self.mousedownpos.x() >= 776)) and self.mousedownpos.y() < 24: self.app.quit() def mouseReleaseEvent(self, event): self.mousedown = False def mouseMoveEvent(self, event): if event.buttons() == Qt.LeftButton: pos = event.pos() frameGm = self.frameGeometry() self.setGeometry(frameGm.x() + pos.x() - self.mousedownpos.x(), frameGm.y() + pos.y() - self.mousedownpos.y(), 800, 600) def change_image(self, image): name, lang = image.split('_', 1) if lang == 'en': img = os.path.join(base_path, 'images', '{}.png'.format(name)) else: img = os.path.join(base_path, 'images', lang.upper(), '{}.png'.format(name)) if not os.path.exists(img): return self.pixmap = QPixmap(img) self.label.setPixmap(self.pixmap) class Niko(QWidget): def __init__(self, *args, **kwargs): self.x, self.y, self.start_y, self.app, self.thread = kwargs['x'], kwargs['y'], kwargs['y'], kwargs['app'], kwargs['thread'] self.screen_height = kwargs['screen_height'] del kwargs['x'], kwargs['y'], kwargs['screen_height'], kwargs['app'], kwargs['thread'] super().__init__(*args, **kwargs) self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint) self.setAttribute(Qt.WA_TranslucentBackground) self.setMinimumSize(48, 64) self.setMaximumSize(48, 64) self.label = QLabel(self) self.frames = [QPixmap(os.path.join(base_path, 'images', 'niko{}.png'.format(n))) for n in range(1,4)] self.label.setPixmap(self.frames[1]) self.show() self.thread.start() def getFrame(self): if ((self.y - self.start_y) % 32 >= 16): return 1 if ((self.y - self.start_y) % 64 >= 32): return 0 else: return 2 def update(self): self.label.setPixmap(self.frames[self.getFrame()]) self.y += 2 if self.y > self.screen_height: self.app.quit() return elif self.y > self.screen_height - 64: self.setMinimumSize(48, self.screen_height - self.y) self.resize(48, self.screen_height - self.y) self.move(self.x, self.y) if __name__ == '__main__': app = QApplication(sys.argv) if len(sys.argv) == 3: # "Niko-leaves-the-screen" mode x, y = int(sys.argv[1]), int(sys.argv[2]) thread = AnimationTimer() niko = Niko(x = x, y = y, screen_height = app.primaryScreen().size().height(), app = app, thread = thread) thread.next_frame.connect(niko.update) else: # Author's Journal mode journal = Journal(app = app) save_path = os.path.join(documents_path, 'Oneshot', 'save_progress.oneshot') if os.path.exists(save_path): with open(save_path, 'rb') as save: save.seek(-8, os.SEEK_END) lang = save.read().decode('utf-8') lang = lang[lang.find('[') + 1:lang.find(']')] if lang == 'en_US': lang = 'en' journal.change_image('save_' + lang) else: thread = WatchPipe() thread.change_image.connect(journal.change_image) thread.start() pipe_file = open(pipe_path, "w+") pipe_file.close() app.exec_() try: os.remove(pipe_path) except: # Most likely due to the file being in use. Ignore. pass