gryf/pygtktalog
1
0
Fork 0
mirror of https://github.com/gryf/pygtktalog.git synced 2025-12-17 11:30:19 +01:00
Code Issues Projects Releases Wiki Activity
Files
4b3a2b0e2eb5e5d276040256818e12cd80b4ffad
pygtktalog/moviedb.py
gryf 6c9116b143 * Change DB engine to sqlite. 
* Removed login stuff.
 * Improvements.
2006-11-25 08:13:51 +00:00

892 lines
33 KiB
Python
Executable File
Raw Blame History

#!/usr/bin/python
# -*- coding: iso8859-2 -*-
"""
MovieDB 0.8
features:
- szukanie po tytule filmu/serii
- szukanie po nazwie pliku
todo:
- dodatkowe pozycje w menu w zależności od uprawnień
- edycja plików
- edycja tytułów
- edycja innych rzeczy
"""
# TODO: dodać okna/dialogi dodawania/edycji tytułu + menu kontekstowe
# TODO: dodać okna/dialogi dodawania/edycji pliku + menu kontekstowe
# TODO: dodac tablicę w bazie, słownikującą plyty_asarray, lang (sub, dub), nazwy alternatywne triggery do nich
# TODO: opanować dodawanie/usuwanie pozycji w menu głównego
#{{{ podstawowe importy i sprawdzenia
import sys
import os
WRKDIR = sys.argv[0:][0].split('moviedb2.py')[0]
if WRKDIR[0] != '/':
"""ścieżka nie jest absolutna"""
WRKDIR = os.getcwd()+"/"+WRKDIR
try:
path = os.environ['HOME']
except:
path = "/tmp"
try:
# przeczytaj plik, o ile istnieje
f = open("%s/.moviedb" % path,"rw")
zpliku = f.read()
f.close()
p = {}
zpliku = zpliku.split("\n")
for i in zpliku:
i = i.split("\t")
p[i[0]] = i[1]
if len(p) == 5:
DB=p['db']
CD=p['cdrom']
EPORTXLS=p['exportxls']
else:
DB="moviedb.db"
CD="/mnt/cdrom"
EPORTXLS=False
except:
# w przeciwnym przypadku przjmij wartości domyślne
DB="moviedb.db"
CD="/mnt/cdrom"
EPORTXLS=False
try:
import pygtk
#tell pyGTK, if possible, that we want GTKv2
pygtk.require("2.0")
except:
#Some distributions come with GTK2, but not pyGTK
pass
try:
import gtk
import gtk.glade
except:
print "You need to install pyGTK or GTKv2 ",
print "or set your PYTHONPATH correctly."
print "try: export PYTHONPATH=",
print "/usr/local/lib/python2.2/site-packages/"
sys.exit(1)
#now we have both gtk and gtk.glade imported
#Also, we know we are running GTK v2
try:
from pysqlite2 import dbapi2 as sqlite
except:
print "MovieDB uses SQLite DB.\nYou'll need to get it and the python bindings as well.\nhttp://www.sqlite.org\nhttp://initd.org/tracker/pysqlite"
sys.exit(1)
if EPORTXLS==True:
try:
import pyExcelerator
except:
print "You'll need pyExcelerator, if you want to export DB to XLS format.\nhttp://sourceforge.net/projects/pyexcelerator"
sys.exit(1)
#}}}
class SaveAsMDB:
"""pokazuje dialog zapisu(exportu) do pliku""" #{{{
def __init__(self):
self.gladefile = WRKDIR + "/glade/moviedb.glade"
def run(self):
self.savegld = gtk.glade.XML(self.gladefile, "saveasMDB")
self.save = self.savegld.get_widget("saveasMDB")
self.savegld.get_widget("saveasMDB")
self.save.set_do_overwrite_confirmation(True)
self.save.set_title("MovieDB - Save as...")
uri = self.save.set_current_name("moviedb.xls")
self.result = self.save.run()
uri = self.save.get_uri()
self.save.destroy()
return self.result,uri
#}}}
class SelectFolderMDB:
"""pokazuje dialog wskazania katalogu""" #{{{
def __init__(self):
self.gladefile = WRKDIR + "/glade/moviedb.glade"
def run(self):
self.selgld = gtk.glade.XML(self.gladefile, "selectFolderMDB")
self.sel = self.selgld.get_widget("selectFolderMDB")
self.sel.set_title("MovieDB - Select folder")
self.result = self.sel.run()
uri = self.sel.get_uri()
print uri
self.sel.destroy()
return self.result,uri
#}}}
class PrefsMDB:
"""ustawia i zapisuje do pliku ~/.moviedb informacje związane z aplikacją""" #{{{
def __init__(self):
self.prefs = {}
self.gladefile = WRKDIR + "/glade/moviedb.glade"
self.wPrefs = gtk.glade.XML(self.gladefile,"Prefs")
self.wPrefs.get_widget("dbuser").set_text(USER)
self.wPrefs.get_widget("dbpass").set_text(PASS)
self.wPrefs.get_widget("dbhost").set_text(HOST)
self.wPrefs.get_widget("dbname").set_text(DB)
self.wPrefs.get_widget("devdev").set_text(CD)
dic = {"on_prefsCancel_clicked" :self.abandonPrefs,\
"on_prefsSave_clicked" :self.savePrefs,\
"on_selectFolder_clicked" :self.chooseDir}
# podłączenie sygnałów
self.wPrefs.signal_autoconnect(dic)
def abandonPrefs(self, widget):
"""porzuć edycję preferencji"""
self.wPrefs.get_widget("Prefs").destroy()
def savePrefs(self,widget):
"""zapisz ustawienia"""
self.prefs["user"] = self.wPrefs.get_widget("dbuser").get_text()
self.prefs["pass"] = self.wPrefs.get_widget("dbpass").get_text()
self.prefs["host"] = self.wPrefs.get_widget("dbhost").get_text()
self.prefs["db"] = self.wPrefs.get_widget("dbname").get_text()
self.prefs["cdrom"] = self.wPrefs.get_widget("devdev").get_text()
try:
f = open("%s/.moviedb" % path,"w")
f.write("user\t" + self.prefs['user']\
+ "\npass\t" + self.prefs['pass']\
+ "\nhost\t" + self.prefs['host']\
+ "\ndb\t" + self.prefs['db']\
+ "\ncdrom\t" + self.prefs['cdrom']\
)
f.close()
except:
print "Nie można zapisać ustawień!"
# uaktualnij zmienne
USER=self.prefs['user']
PASS=self.prefs['pass']
HOST=self.prefs['host']
DB=self.prefs['db']
CD=self.prefs["cdrom"]
self.wPrefs.get_widget("Prefs").destroy()
def chooseDir(self,widget):
"""wskaż katalog ustawienia"""
sel = SelectFolderMDB()
result,selected = sel.run()
if selected[:8] == 'file:///':
selected = selected[7:]
self.wPrefs.get_widget("devdev").set_text(selected)
# uaktualnij zmienne
DB=selected
#}}}
class StdMSG:
"""pokazuje standartowy dialog - napis i jedne/dwa knefle""" #{{{
def __init__(self, title="", message="", knefli=2):
self.gladefile = WRKDIR + "/glade/moviedb.glade"
self.title = title
self.message = message
self.knefli = knefli
def run(self):
self.msgld = gtk.glade.XML(self.gladefile, "stdMSG")
self.msg = self.msgld.get_widget("stdMSG")
self.msg.set_title(self.title)
self.msgld.get_widget("message").set_markup(self.message)
if self.knefli == 1:
self.msgld.get_widget("cancelbutton2").hide()
self.result = self.msg.run()
self.msg.destroy()
return self.result
#}}}
class AboutMDB:
"""pokazuje prosty dialog "o programie" """ #{{{
def __init__(self):
self.gladefile = WRKDIR + "/glade/moviedb.glade"
def run(self):
self.aboutgld = gtk.glade.XML(self.gladefile, "aboutMDB")
self.about = self.aboutgld.get_widget("aboutMDB")
try:
pixBuf = gtk.gdk.pixbuf_new_from_file(WRKDIR + "/pixmaps/about.png")
self.aboutgld.get_widget("aboutImg").set_from_pixbuf(pixBuf)
except:
pass
self.about.set_title("MovieDB - About")
self.result = self.about.run()
self.about.destroy()
return self.result
#}}}
class DetailsMDB:
"""Pokazuje okno z informacjami wyciągniętymi z DB na podstawie id tytułu""" #{{{
def __init__(self, tid=0):
"""innicjalizacja obiektu"""
gladefile=WRKDIR + "/glade/moviedb.glade"
self.wDetails = gtk.glade.XML(gladefile,"winDetails")
self.wDetails.get_widget("image1").set_from_file(WRKDIR + "/img/notavail.gif")
# pomocnicze zmienne
self.typ = '0'
self.imgIndeks = 1
# inicjalizacja listy z tytułami
import gobject
self.lista=self.wDetails.get_widget("pliki")
self.model=gtk.ListStore(gobject.TYPE_INT,gobject.TYPE_STRING,gobject.TYPE_STRING,\
gobject.TYPE_STRING,gobject.TYPE_STRING,gobject.TYPE_STRING,\
gobject.TYPE_STRING,gobject.TYPE_STRING,gobject.TYPE_STRING,\
gobject.TYPE_STRING, gobject.TYPE_STRING)
self.lista.set_model(self.model)
self.lista.set_headers_visible(True)
columns = [[0,"Id"],[1,u"Płyta"],[2,"Nazwa pliku"],\
[3,"Sub"],[4,"Dub"],[5,"Rozdz."],\
[6,"Video"],[7,"Audio"],[8,"Q"],[9,"Rozm."],[10,'Grupa']]
for col in columns:
renderer=gtk.CellRendererText()
column=gtk.TreeViewColumn(col[1],renderer, text=col[0])
column.set_sort_column_id(col[0])
column.set_resizable(True)
if col[0] != 0:
self.lista.append_column(column)
# inicjalizacja linii statusu
self.sb=self.wDetails.get_widget("statusDet")
self.sbSearchCId = self.sb.get_context_id('detailed res')
self.sbid = 0
# główna robota:
cx = sqlite.connect(WRKDIR+"/db/"+DB,isolation_level=None)
cx.isolation_level = "IMMEDIATE"
c = cx.cursor()
typ = 0
# {{{ SQL: DetailsMDB#1 wyciągnięcie podstawowych informacji o tytule
c.execute("SELECT\
tytul,\
alt,\
data_wydania,\
ilosc_w_serii,\
ilosc_posiadanych,\
nazwa_rodzaju,\
data_zakonczenia,\
a.id_typu\
from\
nazwa_tytulu n\
left join tytul a using(id_tytulu)\
left join rodzaj r using(id_rodzaju)\
where\
n.id_tytulu = %s\
and a.id_typu != %s\
order by\
alt", tid,typ)
#}}}
res = c.fetchall()
# {{{ wielki shit do uzupełnienia formularzyka
if len(res) != 0:
for row in res:
if row[1] == False:
# główny tytuł
if row[7] == 1 or row[7] == 4:
self.typ = 'a'
else:
self.typ = 'f'
self.wDetails.get_widget("tytul").set_text(row[0].decode("iso8859-2"))
self.wDetails.get_widget("winDetails").set_title('MovieDB - '+row[0].decode("iso8859-2"))
# SQL: DetailsMDB#2 pociągnięcie /genre/
c.execute("select nazwa from genre_tytul a left join genre g using(id_genre) where id_tytulu=%s order by nazwa",tid)
gen = c.fetchall()
for g in gen:
if (self.wDetails.get_widget("rodzaj").get_text()) == '':
self.wDetails.get_widget("rodzaj").set_text(g[0])
else:
self.wDetails.get_widget("rodzaj").set_text(self.wDetails.get_widget("rodzaj").get_text()+", "+g[0])
if (self.wDetails.get_widget("rodzaj").get_text()) == '':
self.wDetails.get_widget("rodzaj").hide()
self.wDetails.get_widget("label17").hide()
if row[2]==None:
dadaod = 'unknown'
else:
dadaod = row[2].strftime("%Y-%m-%d")
if row[6]==None:
dadado = 'unknown'
else:
dadado = row[6].strftime("%Y-%m-%d")
if dadaod == dadado:
self.wDetails.get_widget("label15").set_markup("<b>Data wydania:</b>")
self.wDetails.get_widget("data").set_text(dadaod)
else:
self.wDetails.get_widget("data").set_text(dadaod+"/"+dadado)
if row[5] != None:
self.wDetails.get_widget("kategoria").set_text(row[5])
else:
self.wDetails.get_widget("kategoria").hide()
self.wDetails.get_widget("label12").hide()
if row[3] != None:
self.wDetails.get_widget("ilEpizodow").set_text("%d" % row[3])
else:
self.wDetails.get_widget("ilEpizodow").hide()
self.wDetails.get_widget("label13").hide()
if row[4] != None:
self.wDetails.get_widget("ilPosEpis").set_text("%d" % row[4])
else:
self.wDetails.get_widget("ilPosEpis").hide()
self.wDetails.get_widget("label16").hide()
else:
# tytuły są tytułami alternatywnymi
if (self.wDetails.get_widget("tytAlt").get_text()) == '':
self.wDetails.get_widget("tytAlt").set_text(row[0].decode("iso8859-2"))
else:
self.wDetails.get_widget("tytAlt").set_text(self.wDetails.get_widget("tytAlt").get_text()+",\n"+row[0].decode("iso8859-2"))
# sprawdź czy tutuły alternatywne są puste, jeśli tak, ukryj.
if(self.wDetails.get_widget("tytAlt").get_text()) == '':
self.wDetails.get_widget("label11").hide()
self.wDetails.get_widget("tytAlt").hide()
#}}}
# {{{ opanuj obrazki:
try:
pixBuf = gtk.gdk.pixbuf_new_from_file(WRKDIR + "/img/%c%d_1.jpg" % (self.typ,tid))
self.wDetails.get_widget("image1").set_from_pixbuf(pixBuf)
# podłącz sygnały do guzików, wyłącznie w przypadku istnienia obrazków
# "on button1_clicked" : (self.button1_clicked, arg1,arg2)
lista = {"on_popBut_clicked" : (self.prevPict, self.typ, tid),\
"on_nasBut_clicked" : (self.nextPict, self.typ, tid)}
self.wDetails.signal_autoconnect(lista)
except:
self.wDetails.get_widget("image1").set_from_file(WRKDIR + "/img/notavail.gif")
self.wDetails.get_widget("nasBut").hide()
self.wDetails.get_widget("popBut").hide()
self.wDetails.get_widget("hseparator1").hide()
#}}}
# {{{ pokaż listę plików
c = cx.cursor()
# SQL: DetailsMDB#3 pokaż listę plików
c.execute("SELECT\
p.id_pliku,\
aktywny,\
nazwa_pliku,\
format,\
rozdz,\
vcodec,\
acodec,\
rozmiar,\
jakosc,\
nr_plyty,\
nr_pudelka,\
lang_asarray(p.id_pliku::text,'sub'::text),\
lang_asarray(p.id_pliku::text,'dub'::text),\
gp.short\
from\
plik p\
left join plyta y using(id_pliku)\
left join grupa_plik g using(id_pliku)\
left join grupa gp using(id_grupy)\
where\
id_tytulu=%s", tid)
res = c.fetchall()
# pokazanie liości wyników a linii stratusu
if self.sbid != 0:
"""jeśli jest jakiś mesedż, usuń go"""
self.sb.remove(self.sbSearchCId, self.sbid)
liczba_wynikow = u"Ilość plików należących do tytułu: %d" % len(res)
self.sbid = self.sb.push(self.sbSearchCId, liczba_wynikow)
if len(res)>0:
# pokazanie listy
self.model.clear()
for i in res:
self.insert_row(self.model,i[0],"%s/%s" % (i[9],i[10]),i[2].decode("iso8859-2"),i[11],i[12],i[4],i[5],i[6],i[8],i[7],i[13])
#model.set_sort_column_id(1,gtk.SORT_ASCENDING)
#}}}
# SYGNAŁY
def prevPict(self, widget, typ, tid):
"""pokaż poprzedni obrazek, jeśli jest"""
if self.imgIndeks > 1:
self.imgIndeks-=1
try:
pixBuf = gtk.gdk.pixbuf_new_from_file(WRKDIR + "/img/%c%d_%d.jpg" % (typ,tid,self.imgIndeks))
self.wDetails.get_widget("image1").clear()
self.wDetails.get_widget("image1").set_from_pixbuf(pixBuf)
except:
pass
def nextPict(self, widget, typ, tid):
"""pokaż kolejny obrazek, jeśli jest"""
if self.imgIndeks < 5:
self.imgIndeks+=1
try:
pixBuf = gtk.gdk.pixbuf_new_from_file(WRKDIR + "/img/%c%d_%d.jpg" % (typ,tid,self.imgIndeks))
self.wDetails.get_widget("image1").clear()
self.wDetails.get_widget("image1").set_from_pixbuf(pixBuf)
except:
pass
# funkcje pomocnicze
def insert_row(self,model,pid,plyta,nazwa_pliku,sub,dub,rozdz,vid,aud,q,rozm,grup):
myiter=model.insert_after(None,None)
model.set_value(myiter,0,pid)
model.set_value(myiter,1,plyta)
model.set_value(myiter,2,nazwa_pliku)
model.set_value(myiter,3,sub)
model.set_value(myiter,4,dub)
model.set_value(myiter,5,rozdz)
model.set_value(myiter,6,vid)
model.set_value(myiter,7,aud)
model.set_value(myiter,8,q)
model.set_value(myiter,9,rozm)
model.set_value(myiter,10,grup)
#}}}
class MovieDB:
"""pokazująca główne okno aplikacji, pozwalająca na przeszukiwanie wśród plików i tytułów i wyświetlająca wynik wyszukiwania""" #{{{
def __init__(self,dbfile=None):
self.gladefile=WRKDIR + "/glade/moviedb.glade"
self.wTree=gtk.glade.XML(self.gladefile,"mainWin")
# sygnały:
dic = {"on_mainWin_destroy" :(gtk.main_quit),\
"on_quit1_activate" :(gtk.main_quit),\
"on_szukaj_clicked" :(self.searchdb),\
"on_szukPat_activate" :(self.searchdb),\
"on_lista_row_activated" :(self.szczegoly),\
"on_about1_activate" :self.oprogramie,\
"on_cut1_activate" :self.showPrefs,\
"on_save_as1_activate" :self.exportToXLS}
# podłączenie sygnałów
self.wTree.signal_autoconnect(dic)
# inicjalizacja combo boxów
self.treeview=self.wTree.get_widget("pattMatch").set_active(0)
self.treeview=self.wTree.get_widget("typ").set_active(0)
# inicjalizacja listy z tytułami
import gobject
self.treeview=self.wTree.get_widget("lista")
self.treemodel=gtk.TreeStore(gobject.TYPE_INT,gobject.TYPE_STRING,gobject.TYPE_INT,\
gobject.TYPE_INT,gobject.TYPE_STRING,gobject.TYPE_STRING,\
gobject.TYPE_STRING)
self.treeview.set_model(self.treemodel)
self.treeview.set_headers_visible(True)
columns = [[0,"Id"],[1,u"Tytuł"],[2,u"Ilość"],[3,"Posiadanych"],[4,"Rok"],[5,"Typ"],[6,"Plyty"]]
for col in columns:
renderer=gtk.CellRendererText()
column=gtk.TreeViewColumn(col[1],renderer, text=col[0])
column.set_sort_column_id(col[0])
column.set_resizable(True)
if col[0] != 0:
self.treeview.append_column(column)
# inicjalizacja linii statusu
self.sb=self.wTree.get_widget("statusbar")
self.sbSearchCId = self.sb.get_context_id('search res')
self.sbid = 0
return
def insert_row(self,model,parent,tid,tytul,il,pos,rok,typ,plyty):
myiter=model.insert_after(parent,None)
model.set_value(myiter,0,tid)
model.set_value(myiter,1,tytul)
model.set_value(myiter,2,il)
model.set_value(myiter,3,pos)
model.set_value(myiter,4,rok)
model.set_value(myiter,5,typ)
model.set_value(myiter,6,plyty)
return myiter
#####CALLBACKS
def szczegoly(self,treeview,path,view_column):
"""pokaż szczegóły w nowym oknie.""" # {{{
win = DetailsMDB(treeview.get_model().get_value(treeview.get_model().get_iter(path),0))
#}}}
def oprogramie(self, widget):
"""Pokazuje dialog about""" #{{{
about = AboutMDB()
result = about.run()
#}}}
def showPrefs(self, widget):
"""Pokazuje okno preferencji""" #{{{
prefs = PrefsMDB()
#}}}
def searchdb(self,widget):
"""ułóż zapytanie i wypluj wyniki do treeview""" #{{{
if self.wTree.get_widget("pattMatch").get_active() == 3:
pattern=self.wTree.get_widget("szukPat").get_text()
elif self.wTree.get_widget("pattMatch").get_active() == 2:
pattern="%" + self.wTree.get_widget("szukPat").get_text()
elif self.wTree.get_widget("pattMatch").get_active() == 1:
pattern=self.wTree.get_widget("szukPat").get_text() + "%"
else:
pattern="%" + self.wTree.get_widget("szukPat").get_text() + "%"
cx = sqlite.connect(WRKDIR+"/db/"+DB,isolation_level=None)
cx.isolation_level = "IMMEDIATE"
c = cx.cursor()
if self.wTree.get_widget("czalt").get_active():
klon = 'f'
else:
klon = 't'
# szukaj wg typu:
# NOTE: daj tu to, co potrzeba
anime,filmy,kreskowki,xxx,typ = self.getType()
if self.wTree.get_widget("radn").get_active():
# SQL: zapytanie zbierające tytuły filmów po kluczu nazw plików
# TODO: zrobić tablicę przechowującą nr płyt (najlepiej triggerze)
"""
c.execute("SELECT distinct\
a.id_tytulu,\
n.tytul,\
alt,\
data_wydania,\
case\
when\
ilosc_w_serii is null then 0\
else\
ilosc_w_serii\
end,\
case\
when\
ilosc_posiadanych is null then 0\
else\
ilosc_posiadanych\
end,\
nazwa_typu,\
1\
from\
nazwa_tytulu as n\
left join tytul a using(id_tytulu)\
left join typ as t using(id_typu)\
left join plik p using(id_tytulu)\
where\
(alt = 'f' or alt = %s)\
and nazwa_pliku ilike %s\
and id_typu in (%s, %s, %s, %s)\
and id_typu != %s\
order by\
id_tytulu, alt",(klon,pattern,anime,xxx,filmy,kreskowki,typ))
"""
c.execute("SELECT distinct\
a.id_tytulu,\
n.tytul,\
alt,\
data_wydania,\
case\
when\
ilosc_w_serii is null then 0\
else\
ilosc_w_serii\
end,\
case\
when\
ilosc_posiadanych is null then 0\
else\
ilosc_posiadanych\
end,\
nazwa_typu,\
1\
from\
nazwa_tytulu as n\
left join tytul a using(id_tytulu)\
left join typ as t using(id_typu)\
left join plik p using(id_tytulu)")
else:
# SQL: zapytanie zbierające tytuły filmów
"""
c.execute("SELECT \
a.id_tytulu,\
n.tytul,\
alt,\
data_wydania,\
case\
when\
ilosc_w_serii is null then 0\
else\
ilosc_w_serii\
end,\
case\
when\
ilosc_posiadanych is null then 0\
else\
ilosc_posiadanych\
end,\
nazwa_typu,\
1\
from\
nazwa_tytulu as n\
left join tytul a using(id_tytulu)\
left join typ as t using(id_typu)\
where\
(alt = 'f' or alt = %s)\
and tytul ilike %s\
and id_typu in (%s, %s, %s, %s)\
and id_typu != %s\
order by\
id_tytulu, alt",(klon,pattern,anime,xxx,filmy,kreskowki,typ))
"""
c.execute("SELECT distinct\
a.id_tytulu,\
n.tytul,\
alt,\
data_wydania,\
ilosc_w_serii,\
ilosc_posiadanych,\
nazwa_typu,\
1\
from\
nazwa_tytulu as n\
left join tytul as a on a.id_tytulu=n.id_tytulu\
left join typ as t using (id_typu)\
left join plik as p on p.id_tytulu=n.id_tytulu\
where\
tytul like '%s'" % pattern)
res = c.fetchall()
cx.close()
# pokazanie liości wyników a linii stratusu
if self.sbid != 0:
"""jeśli jest jakiś mesedż, usuń go"""
self.sb.remove(self.sbSearchCId, self.sbid)
liczba_wynikow = u"Ilość znalezionych rekordów: %d" % len(res)
self.sbid = self.sb.push(self.sbSearchCId, liczba_wynikow)
if len(res)>0:
# pokazanie listy
model=self.treemodel
self.treemodel.clear()
if self.wTree.get_widget("showAsTree").get_active():
ido=0
for i in res:
if i[2] == True and ido == i[0]:
self.insert_row(model,iterobj,i[0],i[1],i[4],i[5],i[3],i[6],i[7])
else:
iterobj = self.insert_row(model,None,i[0],i[1],i[4],i[5],i[3],i[6],i[7])
ido = i[0]
else:
for i in res:
self.insert_row(model,None,i[0],i[1],i[4],i[5],i[3],i[6],i[7])
#model.set_sort_column_id(1,gtk.SORT_ASCENDING)
# pokazanie wyników w tabelce
self.wTree.get_widget("scrListy").show()
self.treeview.expand_all()
else:
self.wTree.get_widget("scrListy").hide()
#}}}
def getType(self):
"""zdejmuje typ dla zapytania SQL"""
typ = self.wTree.get_widget("typ").get_active()
if typ == 4:
anime = 4
filmy = 4
kreskowki = 4
xxx = 4
elif typ == 3:
anime = 3
filmy = 3
kreskowki = 3
xxx = 3
elif typ == 2:
anime = 2
filmy = 2
kreskowki = 2
xxx = 2
elif typ == 1:
anime = 1
filmy = 1
kreskowki = 1
xxx = 1
else:
anime = 1
filmy = 2
kreskowki = 3
xxx = 4
typ=0
return anime,filmy,kreskowki,xxx,typ
def exportToXLS(self,widget):
"""eksportuj bazę do arkusza xls""" #{{{
fIter = self.treemodel.get_iter_first()
ajdiki = "(0"
if fIter!=None:
ids = []
ids.append(self.treemodel.get_value(fIter,0))
nIter = self.treemodel.iter_next(fIter)
while nIter!=None:
ids.append(self.treemodel.get_value(nIter,0))
nIter = self.treemodel.iter_next(nIter)
ajl = []
for i in set(ids):
ajl.append(i)
ajdiki+=",%s" % i
else:
pass
ajdiki+=")"
idnum = len(ajl)
sa = SaveAsMDB()
result,uri = sa.run()
if (result == gtk.RESPONSE_OK):
# open(uri[7:],"w")
import pyExcelerator as X
w = X.Workbook()
X.UnicodeUtils.DEFAULT_ENCODING = 'cp1250'
anime,filmy,kreskowki,xxx,typ = self.getType()
cx = sqlite.connect(user=USER, password=PASS, host=HOST, database=DB, client_encoding="iso8859-2")
c = cx.cursor()
sql = "select\
p.id_pliku,\
case when nr_plyty is null then 'brak' else nr_plyty end,\
nazwa_pliku,\
rozdz,\
vcodec,\
acodec,\
jakosc,\
rozmiar,\
p.id_tytulu,\
to_char (p.data_dodania,'YYYY-MM-DD HH24:MM:SS'),\
lang_asarray(p.id_pliku::text,'dub'),\
lang_asarray(p.id_pliku::text,'sub')\
from\
plik p\
left join plyta l using(id_pliku)\
left join tytul t using(id_tytulu)\
where\
p.aktywny is true\
and\
t.id_typu in (%s, %s, %s, %s)" % (anime,filmy,kreskowki,xxx)
sql+=" and\
t.id_typu != %s" % typ
sql+=" and\
p.id_tytulu in ("+ ", ".join(["%s"] * idnum) + ")\
order by\
nazwa_pliku"
c.execute(sql, ajl)
pliki = c.fetchall()
row = 0
sheet = w.add_sheet("Anime")
# nagłówki
sheet.write(row, 0, u"tytuł anime".encode("cp1250"))
sheet.write(row, 1, u"nr płyty".encode("cp1250"))
sheet.write(row, 2, "nazwa pliku".encode("cp1250"))
sheet.write(row, 3, "dub".encode("cp1250"))
sheet.write(row, 4, "sub".encode("cp1250"))
sheet.write(row, 5, "rozdz".encode("cp1250"))
sheet.write(row, 6, "vcodec".encode("cp1250"))
sheet.write(row, 7, "acodec".encode("cp1250"))
sheet.write(row, 8, u"jakość".encode("cp1250"))
sheet.write(row, 9, "rozmiar pliku".encode("cp1250"))
sheet.write(row, 10, "typ".encode("cp1250"))
sheet.write(row, 11, "data".encode("cp1250"))
sheet.write(row, 12, u"ilość odc.".encode("cp1250"))
sheet.write(row, 13, u"ilość posiadanych".encode("cp1250"))
sheet.write(row, 14, "data dodania pliku".encode("cp1250"))
row+=1
idanime="0"
for plik in pliki:
if plik[9]!=idanime:
sql="SELECT\
tytul,\
alt,\
nazwa_rodzaju,\
case when data_wydania is null then 'brak' else to_char (data_wydania,'YYYY-MM-DD') end,\
data_zakonczenia,\
ilosc_w_serii,\
ilosc_posiadanych\
from\
nazwa_tytulu as n\
left join tytul a using(id_tytulu)\
left join rodzaj r using(id_rodzaju)\
where\
a.id_tytulu=%s\
and\
a.id_typu in (%s, %s, %s, %s)\
and\
a.id_typu != %s\
and\
alt is false\
order by\
tytul" % (plik[8],anime,filmy,kreskowki,xxx,typ)
c.execute(sql)
tytuly = c.fetchone()
print tytuly[0]
sheet.write(row, 0, tytuly[0].decode("iso8859-2").encode("cp1250"))
sheet.write(row, 1, plik[1].encode("cp1250"))
sheet.write(row, 2, plik[2].decode("iso8859-2").encode("cp1250"))
sheet.write(row, 3, plik[10].encode("cp1250"))
sheet.write(row, 4, plik[11].encode("cp1250"))
sheet.write(row, 5, plik[3].encode("cp1250"))
sheet.write(row, 6, plik[4].encode("cp1250"))
sheet.write(row, 7, plik[5].encode("cp1250"))
sheet.write(row, 8, plik[6])
sheet.write(row, 9, plik[7])
sheet.write(row, 10, "typ".encode("cp1250"))
sheet.write(row, 11, tytuly[3])
sheet.write(row, 12, tytuly[5])
sheet.write(row, 13, tytuly[6])
sheet.write(row, 14, plik[9].encode("cp1250"))
row+=1
w.save(uri[7:])
cx.close()
#}}}
#}}}
# NOTE: koniec deklaracji klas. główny program:
try:
app=MovieDB(sys.argv[1])
except:
app=MovieDB()
try:
gtk.main()
except KeyboardInterrupt:
gtk.main_quit
Reference in New Issue View Git Blame Copy Permalink