revpi/revpipycontrol
1
0
Fork 0
mirror of https://github.com/naruxde/revpipycontrol.git synced 2025-11-08 15:43:52 +01:00
Code Issues Releases Wiki Activity

upload eingebaut (noch mit tar)

Browse Source
This commit is contained in:
Sven Sager
2017-03-10 16:24:54 +01:00
parent fb3de16ef7
commit d5c01c2825
2 changed files with 191 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
revpipycontrol/revpiprogram.py
View File

@@ -5,11 +5,12 @@
# (c) Sven Sager, License: LGPLv3
#
# -*- coding: utf-8 -*-
import os
import tarfile
import tkinter
import tkinter.filedialog as tkfd
import tkinter.messagebox as tkmsg
from os import makedirs
import zipfile
from tempfile import mktemp
from xmlrpc.client import Binary
@@ -46,9 +47,9 @@ class RevPiProgram(tkinter.Frame):
self.columnconfigure(0, weight=1)
cpad = {"padx": 4, "pady": 2}
cpade = {"padx": 4, "pady": 2, "sticky": "e"}
# cpade = {"padx": 4, "pady": 2, "sticky": "e"}
cpadw = {"padx": 4, "pady": 2, "sticky": "w"}
cpadwe = {"padx": 4, "pady": 2, "sticky": "we"}
# cpadwe = {"padx": 4, "pady": 2, "sticky": "we"}
# Gruppe Programm
prog = tkinter.LabelFrame(self)
@@ -64,7 +65,7 @@ class RevPiProgram(tkinter.Frame):
self.var_typeup = tkinter.StringVar(prog)
self.lst_typedown = ["Dateien", "Zip Archiv", "TGZ Archiv"]
self.lst_typeup = ["Datei", "Ordner", "Zip Archiv", "TGZ Archiv"]
self.lst_typeup = ["Dateien", "Ordner", "Zip Archiv", "TGZ Archiv"]
self.var_typedown.set(self.lst_typedown[0])
self.var_typeup.set(self.lst_typeup[0])
@@ -99,6 +100,12 @@ class RevPiProgram(tkinter.Frame):
opt.grid(column=1, row=r, **cpad)
r = 3
ckb = tkinter.Checkbutton(prog)
ckb["text"] = "vorher alles im Uploadverzeichnis löschen"
ckb["variable"] = self.var_cleanup
ckb.grid(column=0, row=r, columnspan=2, **cpadw)
r = 4
self.ckb_picup = tkinter.Checkbutton(prog)
self.ckb_picup["text"] = "enthält piCtory Konfiguration"
self.ckb_picup["variable"] = self.var_picup
@@ -108,12 +115,6 @@ class RevPiProgram(tkinter.Frame):
btn["text"] = "Upload"
btn.grid(column=1, row=r, **cpad)
r = 4
ckb = tkinter.Checkbutton(prog)
ckb["text"] = "vorher alles im Uploadverzeichnis löschen"
ckb["variable"] = self.var_cleanup
ckb.grid(column=0, row=r, columnspan=2, **cpadw)
# Gruppe piCtory
picto = tkinter.LabelFrame(self)
picto.columnconfigure(0, weight=1)
@@ -162,12 +163,20 @@ class RevPiProgram(tkinter.Frame):
self.ckb_picdown["state"] = "normal"
def _evt_optup(self, text=""):
if self.lst_typeup.index(self.var_typeup.get()) == 0:
if self.lst_typeup.index(self.var_typeup.get()) <= 1:
self.var_picup.set(False)
self.ckb_picup["state"] = "disable"
else:
self.ckb_picup["state"] = "normal"
def _loaddefault(self):
# TODO: letzte Einstellungen laden
pass
def _savedefaults(self):
# TODO: Einstellungen sichern
pass
def getpictoryrsc(self):
fh = tkfd.asksaveasfile(
mode="wb", parent=self.master,
@@ -221,7 +230,6 @@ class RevPiProgram(tkinter.Frame):
fh.close()
def setpictoryrsc(self):
print("setpictoryrsc")
fh = tkfd.askopenfile(
mode="rb", parent=self.master,
title="piCtory Datei öffnen...",
@@ -267,15 +275,15 @@ class RevPiProgram(tkinter.Frame):
def plcdownload(self):
tdown = self.lst_typedown.index(self.var_typedown.get())
fh = None
dir = ""
dirselect = ""
if tdown == 0:
# Ordner
dir = tkfd.askdirectory(
dirselect = tkfd.askdirectory(
parent=self.master, title="Verzeichnis zum Ablegen",
mustexist=False, initialdir=self.revpi)
if type(dir) == str and dir != "":
if type(dirselect) == str and dirselect != "":
fh = open(mktemp(), "wb")
elif tdown == 1:
@@ -300,18 +308,28 @@ class RevPiProgram(tkinter.Frame):
if fh is not None:
if tdown == 1:
plcfile = self.xmlcli.plcdownload("zip")
plcfile = self.xmlcli.plcdownload(
"zip", self.var_picdown.get())
else:
plcfile = self.xmlcli.plcdownload("tar")
plcfile = self.xmlcli.plcdownload(
"tar", self.var_picdown.get())
try:
fh.write(plcfile.data)
# Optional entpacken
if tdown == 0:
fh.close()
makedirs(dir, exist_ok=True)
os.makedirs(dirselect, exist_ok=True)
fh_pack = tarfile.open(fh.name)
fh_pack.extractall(dir)
# Unterverzeichnis streichen
rootname = ""
for taritem in fh_pack.getmembers():
print(rootname)
if not taritem.name == "revpipyload":
taritem.name = taritem.name.replace("revpipyload/", "")
fh_pack.extract(taritem, dirselect)
fh_pack.close()
except:
@@ -330,14 +348,161 @@ class RevPiProgram(tkinter.Frame):
finally:
fh.close()
print("plcdownload", tdown)
def plcupload(self):
tup = self.lst_typeup.index(self.var_typeup.get())
fh = None
dirselect = ""
print("plcupload", tup)
if tup == 0:
# Datei
fileselect = tkfd.askopenfilenames(
parent=self.master, title="Pythonprogramm übertragen...",
filetypes=(("Python", "*.py"), ("All Files", "*.*"))
)
if type(fileselect) == tuple and len(fileselect) > 0:
# Datei als TAR packen
tmpfile = mktemp()
noerr = True
try:
fh_pack = tarfile.open(
name=tmpfile, mode="w:gz", dereference=True)
for file in fileselect:
fh_pack.add(
file, arcname=os.path.basename(file))
except:
noerr = False
tkmsg.showerror(
parent=self.master, title="Fehler",
message="Die Datei konnte für die Übertragung nicht "
"gepackt werden")
finally:
fh_pack.close()
if noerr:
# fh für Versand öffnen
fh = open(tmpfile, "rb")
elif tup == 1:
# Ordner
dirselect = tkfd.askdirectory(
parent=self.master, title="Verzeichnis zum Hochladen",
mustexist=True, initialdir=self.revpi)
if type(dirselect) == str and dirselect != "":
# Ordner als TAR packen
tmpfile = mktemp()
noerr = True
try:
fh_pack = tarfile.open(
name=tmpfile, mode="w:gz", dereference=True)
fh_pack.add(
#dirselect, arcname=os.path.basename(dirselect))
dirselect, arcname="")
except:
noerr = False
tkmsg.showerror(
parent=self.master, title="Fehler",
message="Der Ordner konnte für die Übertragung nicht "
"gepackt werden")
finally:
fh_pack.close()
if noerr:
# fh für Versand öffnen
fh = open(tmpfile, "rb")
elif tup == 2:
# Zip
fh = tkfd.askopenfile(
mode="rb", parent=self.master,
title="Zip-Archive übertragen...",
initialfile=self.revpi + ".zip",
filetypes=(("Zip Archiv", "*.zip"), ("All Files", "*.*"))
)
if not zipfile.is_zipfile(fh.name):
# Zipdatei prüfen
tkmsg.showerror(
parent=self.master, title="Fehler",
message="Die angegebene Datei ist kein ZIP-Archiv.")
fh.close()
fh = None
elif tup == 3:
# TarGz
fh = tkfd.askopenfile(
mode="rb", parent=self.master,
title="TarGz-Archiv übertragen...",
initialfile=self.revpi + ".tar.gz",
filetypes=(("Tar Archiv", "*.tar.gz"), ("All Files", "*.*"))
)
if not tarfile.is_tarfile(fh.name):
# Zipdatei prüfen
tkmsg.showerror(
parent=self.master, title="Fehler",
message="Die angegebene Datei ist kein TAR-Archiv.")
fh.close()
fh = None
# Wenn kein fh existiert abbrachen
if fh is None:
return False
# Vor Übertragung aufräumen wenn ausgewählt
if self.var_cleanup.get() and not self.xmlcli.plcuploadclean():
tkmsg.showerror(
parent=self.masger, title="Fehler",
message="Beim Löschen der Dateien auf dem Revolution Pi ist "
"ein Fehler aufgetreten.")
return False
# Flag setzen, weil ab hier Veränderungen existieren
self.uploaded = True
# piControlReset abfragen
ask = False
if self.var_picup.get():
ask = tkmsg.askyesno(
parent=self.master, title="Frage",
message="Sie laden eine piCtory Konfiguration mit hoch. \n"
"Soll nach dem Hochladen ein reset am piControl "
"Treiber durchgeführt werden?",
)
# TODO: Fehlerabfang bei Dateilesen
xmldata = Binary(fh.read())
ec = self.xmlcli.plcupload(xmldata, self.var_picup.get(), ask)
if ec == 0:
tkmsg.showinfo(
parent=self.master, title="Erfolgreich",
message="Die Übertragung war erfolgreich.")
elif ec > 0:
tkmsg.showwarning(
parent=self.master, title="Warnung",
message="Die Übertragung war erfolgreich. \n"
"Beim piControl Reset trat allerdings ein Fehler auf!")
elif ec == -1:
tkmsg.showerror(
parent=self.master, title="Fehler",
message="Der Revoluton Pi konnte die übertragene Datei nicht "
"verarbeiten.")
elif ec < -1:
tkmsg.showwarning(
parent=self.master, title="Warnung",
message="Die Übertragung war erfolgreich. \n"
"Beim verarbeiten der piCtory Konfiguration trat allerdings "
"ein Fehler auf!")
fh.close()
# Temp-File aufräumen
if tup <= 1:
os.remove(fh.name)
if __name__ == "__main__":
root = tkinter.Tk()

4
revpipycontrol/revpipycontrol.py
View File

@@ -155,12 +155,16 @@ class RevPiPyControl(tkinter.Frame):
def plcoptions(self):
win = tkinter.Toplevel(self)
revpioption.RevPiOption(win, self.cli)
win.focus_set()
win.grab_set()
self.wait_window(win)
def plcprogram(self):
# TODO: Programfenster
win = tkinter.Toplevel(self)
revpiprogram.RevPiProgram(win, self.cli, self.revpiname)
win.focus_set()
win.grab_set()
self.wait_window(win)
def plcstart(self):