Python图片浏览器

　　上次我爬了n多图片，但是浏览的时候有一个问题。

　　图片浏览器的浏览一般都是按名称排的，而我对图片的命名是按照数字递增的。比如3总是会排在10后面，也就无法快速地浏览图片了。

　　所以，出于方便自己查阅图片，也出于学习，决定做一个自己的图片浏览器。

　　

　　目标：浏览目录，通过滚轮不断显示同一个文件夹下的图片，并自定义排序。

 

　　步骤0：要实现图形界面，我使用wxPython。

　　至于如何安装和简单地使用wxpython，可以到网上检索，一大堆资料。

　　以下步骤默认你已经知道如何生成一个自己的frame。

　　但是wxPython的文档却比较难找到，其中这两个比较好：wxPython里的类、wxPython里的事件。

 

　　步骤1：浏览目录。

　　这个功能就是类似于打开“我的电脑”，然后不断地进入文件夹和返回。

　　通过几种尝试，我决定使用listbox。listbox的使用可以通过这篇文章了解。

　　我初始化一个app。用一个frame实现目录的功能，其上只有一个listbox；用另一个frame实现图片展示的功能，两个frame通过app进行信息的传递。

for _dir in os.listdir(dir):
    #do something

　　其中像os.path.split()、os.path.splitext()、os.path.isdir()等，都是很常用的一些方法。

　　显示目录就是一个不断地获取你选择的目录，进入目录，读取其下目录，清空listbox，显示目录，更改工作路径的过程。

　　显示目录的时候，自定义排序功能就来了。通过对图片名称进行处理，转为数字，排序，然后再重新组装回去，从而达到按数字递增的效果。

self.list.Bind(wx.EVT_LISTBOX_DCLICK, self.OnDClick)

　　同时通过以上方法为listbox绑定了双击事件。若是双击目录则进入目录，否则显示通过app展示图片。

　　同时这个frame需要具备两个方法，就是获取上一张或下一张图片，为了后续app的调用。

 

　　步骤2：显示图片

　　这个图片展示一开始感觉挺麻烦的，但是弄懂了之后就很简单了。

　　我用一个frame展示图片。

　　frame上面空白，有一个wx.StaticBitmap，之后显示图片的时候只需要往这个staticbitmap写bitmap就可以了。一开始我是不断地新建staticbitmap，导致了一些可以看到但是那时候不知道为什么的原因。

　　然后在这个frame上检测鼠标滚轮事件，通过向上或向下滚轮调用app的GetNextImage和GetPreImage方法并将获得的图片显示出来。

　　还有图片的大小，我先规定了一个最大值和最小值，将图片约束在一定的范围内。

bmp = image.Scale(size[0], size[1]).ConvertToBitmap()
self.bmp.SetSize(size)#bmp是staticbitmap
self.bmp.SetBitmap(bmp)

　　但是一个小窗口看图片很不爽，于是将图片窗口全屏化。

self.ShowFullScreen(True, style=wx.FULLSCREEN_ALL)

　　全屏化就要考虑怎么退出了。我通过按键发送消息，命令窗口关闭或显示。若显示则关闭（其实只是隐藏），若隐藏则显示。注意这个事件要绑定到app上面。

self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)

　　然后图片需要能够放大和缩小。于是我再次通过按键触发。放大或缩小只需要通过改变staticbitmap的最大值并让bmp适应那个size就可以了。

　　由于全屏了，那么需要能够移动图片。移动的时候也只需要移动staticbitmap就行了。

#注意这里要将事件绑定到staticbitmap上面
self.bmp.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
self.bmp.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
self.bmp.Bind(wx.EVT_MOTION, self.OnMotion)

　　

　　至此，大概要点都讲完了，下面是全部代码。仔细查阅会发现一些新的用法，

　　不过这个图片浏览器估计只是够我用，不过其实用起来已经挺爽的了， 全屏的图片展示，还可以随便移动图片，快捷键很方便地退出全屏。

　　里面还有一些不完善的地方需要改进，希望能跟大家多多交流~

　　感谢这期间被我大量参考资料的作者们。

#!/usr/bin/env Python
#coding=utf-8

#filename : PictureBrowser.py
#author   : http://www.cnblogs.com/chenkun24/
#date     : 2012-10-11



import wx
import os
import sys
import string

#你有H盘吗？没有的话在这个初始化函数里修改加载的初始路径
class PBDirFrame(wx.Frame):
    def __init__(self, app):
        wx.Frame.__init__(self, None, -1, "选择文件夹", size=(250,500))

        self.app = app

        #设置字体
        font = wx.Font(12, wx.DEFAULT, wx.NORMAL, wx.NORMAL, False, 'Courier New')
        self.SetFont(font)
        
        #文件夹listbox
        self.list = wx.ListBox(self, -1, (0,0), (200,600), '', wx.LB_SINGLE)
        self.list.Bind(wx.EVT_LISTBOX_DCLICK, self.OnDClick)
        
        #加载当前文件夹
        #curdir = os.getcwd()#在这里修改初始路径，这个是当前工作路径
        curdir = 'H:\\'
        os.chdir(curdir)
        self.LoadDir(curdir)
        
        #绑定事件
        self.Bind(wx.EVT_CLOSE, self.OnClose)


        
        #显示窗口
        self.Show()
    
    def OnClose(self, event):
        self.Destroy()
        self.app.Close()
    
    #listbox双击事件
    def OnDClick(self, event):
        if self.list.GetSelection()==0:#判断是否选择了返回上一层文件夹
            path = os.getcwd()
            pathinfo = os.path.split(path)
            dir = pathinfo[0]
        else:#获得需要进入的下一层文件夹
            dir = self.list.GetStringSelection()
        
        if os.path.isdir(dir):#进入文件夹
            self.LoadDir(dir)
        elif os.path.splitext(dir)[-1]=='.jpg':#显示图片
            self.app.ShowImage(dir)

    #加载文件夹，如果你想定义自己的排序，那么修改这个方法吧~
    def LoadDir(self, dir):
        #不是目录则不进行操作
        if not os.path.isdir(dir):
            return
        
        self.list.Clear()#清空
        self.list.Append('...')#添加返回上一层文件夹标志

        dirs = []
        jpgs = []
        nnjpgs = []
        for _dir in os.listdir(dir):
            if os.path.isdir(dir+os.path.sep+_dir):
                dirs.append(_dir)
            else:
                info = os.path.splitext(_dir)
                if info[-1]=='.jpg':
                    if info[0].isdigit():
                        jpgs.append(string.atoi(info[0]))#转化为数字
                    else:
                        nnjpgs.append(_dir)
        jpgs.sort()
        for _jpgs in jpgs:
            self.list.Append(str(_jpgs)+'.jpg')
        for _nnjpgs in nnjpgs:
            self.list.Append(_nnjpgs)
        for _dirs in dirs:
            self.list.Append(_dirs)

        os.chdir(dir)#设置工作路径

    #获得下一张要显示的图片
    def GetNextImage(self):
        index = self.list.GetSelection()
        i = index
        while i+1<self.list.GetCount():
            i += 1
            if os.path.splitext(self.list.GetString(i))[-1]=='.jpg':
                break
        if i<self.list.GetCount():
            index = i
        self.list.SetSelection(index)
        return self.list.GetStringSelection()

    #获得上一张图片
    def GetPreImage(self):
        index = self.list.GetSelection()
        i = index
        while i-1>0:
            i -= 1
            if os.path.splitext(self.list.GetString(i))[-1]=='.jpg':
                break
        if i>0:
            index = i
        
        self.list.SetSelection(index)
        return self.list.GetStringSelection()


class PBPicFrame(wx.Frame):

    max_width = 400
    max_height = 600

    def __init__(self, app):
        wx.Frame.__init__(self, None, -1, "显示图片", size=(400,400))#, style=wx.SIMPLE_BORDER)

        #是否要移动图片的标志
        self.bmoved = False
        
        self.app = app

        #staticbitmap
        self.bmp = wx.StaticBitmap(self, 0, wx.NullBitmap, (0,0), (400,400))


        self.Bind(wx.EVT_MOUSEWHEEL, self.OnChangeImage)
        self.bmp.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
        self.bmp.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)
        self.bmp.Bind(wx.EVT_MOTION, self.OnMotion)
        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
        
        self.ShowFullScreen(True, style=wx.FULLSCREEN_ALL)
        self.Hide()


    def ShowImage(self, path):
        if os.path.splitext(path)[-1]!='.jpg':
            return
        self.bmppath = path
        image = wx.Image(path, wx.BITMAP_TYPE_JPEG)
        bmp = image.ConvertToBitmap()
        size = self.GetSize(bmp)
        bmp = image.Scale(size[0], size[1]).ConvertToBitmap()
        self.bmp.SetSize(size)
        self.bmp.SetBitmap(bmp)
        self.Show()

    def GetSize(self, bmp):
        width = bmp.GetWidth()
        height = bmp.GetHeight()
        if width>self.max_width:
            height = height*self.max_width/width
            width = self.max_width
        if height>self.max_height:
            width = width*self.max_height/height
            height = self.max_height
        size = width, height
        return size
        

    def OnChangeImage(self, event):
        rotation = event.GetWheelRotation()
        if rotation<0:
            self.app.ShowNextImage()
        else:
            self.app.ShowPreImage()
    
    def OnLeftDown(self, event):
        self.pos = event.GetX(), event.GetY()
        self.bmoved = True

    def OnLeftUp(self, event):
        self.bmoved = False

    def OnMotion(self, event):
        if not self.bmoved:
            return
        pos = event.GetX(), event.GetY()
        dx = pos[0]-self.pos[0]
        dy = pos[1]-self.pos[1]
        pos = self.bmp.GetPosition()
        pos = pos[0]+dx, pos[1]+dy
        self.bmp.SetPosition(pos)

    def OnKeyDown(self, event):
        keycode = event.GetKeyCode()
        if keycode == 49:#数字1放大
            self.SizeUp()
        elif keycode == 50:#数字2缩小
            self.SizeDown()
        event.Skip()#这个貌似很重要，要同时触发app上的快捷键

    def SizeUp(self):
        self.max_width += 50
        self.max_height += 75
        self.ShowImage(self.bmppath)
    def SizeDown(self):
        self.max_width -= 50
        self.max_height -= 75
        self.ShowImage(self.bmppath)

class PBApp(wx.App):
    
    #redirect=False将信息输出到dos界面
    def __init__(self, redirect=False):
        wx.App.__init__(self, redirect)
    
    def OnInit(self):
        
        #显示文件夹列表界面
        self.dirframe = PBDirFrame(self)
        
        #显示图片界面
        self.picframe = PBPicFrame(self)
        
        #绑定事件
        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
        return True

    def ShowImage(self, path):
        #print 'showing app img', path
        self.picframe.ShowImage(path)
        self.picframe.SetFocus()
    
    def ShowNextImage(self):
        path = self.dirframe.GetNextImage()
        self.ShowImage(path)

    def ShowPreImage(self):
        path = self.dirframe.GetPreImage()
        self.ShowImage(path)

    def OnKeyDown(self, event):
        keycode = event.GetKeyCode()
        #print keycode
        if keycode == 27:# ESC键
            #切换图片窗体的显示和隐藏
            if self.picframe.IsShown():
                self.picframe.Hide()
            else:
                self.picframe.Show()
    
    def Close(self):
        self.picframe.Close()
    
    
def main():
    app = PBApp()
    app.MainLoop()

if __name__=='__main__':
    main()