LogarithmPlotter/run.py

"""
 *  Logarithm Graph Creator - Create graphs with logarithm scales.
 *  Copyright (C) 2020  Ad5001
 * 
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 * 
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""

from PySide2.QtWidgets import QApplication, QFileDialog
from PySide2.QtQml import QQmlApplicationEngine, qmlRegisterType
from PySide2.QtCore import Qt, QObject, Signal, Slot, Property

import os
import tempfile
from platform import release as os_release
from json import dumps
from sys import platform

tempfile = tempfile.mkstemp(suffix = '.png')[1]

def get_linux_theme():
    des = {
        "KDE": "fusion", # org.kde.desktop resolves to universal in PySide2.
        "gnome": "default",
        "lxqt": "fusion",
        "mate": "fusion",
    }
    if "XDG_SESSION_DESKTOP" in os.environ:
        return des[os.environ["XDG_SESSION_DESKTOP"]] if os.environ["XDG_SESSION_DESKTOP"] in des else "fusion"
    else:
        # Android
        return "Material"
os.environ["QT_QUICK_CONTROLS_STYLE"] = {
    "linux": get_linux_theme(),
    "freebsd": get_linux_theme(),
    "win32": "universal" if os_release == "10" else "fusion",
    "cygwin": "fusion",
    "darwin": "imagine"
}[platform]

class Helper(QObject):
    @Slot(str, str)
    def write(self, filename, filedata):
        if os.path.exists(os.path.dirname(os.path.realpath(filename))):
            f = open(os.path.realpath(filename), 'w',  -1, 'utf8')
            f.write(filedata)
            f.close()
        
    @Slot(str, result=str)
    def load(self, filename):
        if os.path.exists(os.path.realpath(filename)):
            f = open(os.path.realpath(filename), 'r',  -1, 'utf8')
            data = f.read()
            f.close()
            return data
        return '{}'

    @Slot(result=str)
    def gettmpfile(self):
        global tempfile
        return tempfile
    
    @Slot()
    def copyImageToClipboard(self):
        global tempfile
        # TODO: Better copy system
        os.system("xclip -selection clipboard -t image/png -i " + tempfile)

app = QApplication([])
app.setApplicationName("Logarithmic Plotter")
app.setOrganizationName("Ad5001")
engine = QQmlApplicationEngine()
helper = Helper()
engine.rootContext().setContextProperty("Helper", helper)

engine.addImportPath(os.path.realpath(os.path.join(os.getcwd(), "qml")))
engine.load(os.path.realpath(os.path.join(os.getcwd(), "qml", "LogGraph.qml")))

if not engine.rootObjects():
    print("No root object")
    exit(-1)
app.exec_()

os.remove(tempfile)