import wx
import os
import shutil
import time



class Window(wx.Frame):
    def __init__(self, parent, id):
        self.Desktop = "C:\Users\name\Desktop\\"
        self.Pdesktop = "C:\Users\Public\Desktop\\"
        self.Dfiles = "C:\Desktop_Files\\"
        no_caption = wx.DEFAULT_FRAME_STYLE | wx.STAY_ON_TOP
        wx.Frame.__init__(self, parent, id, title="no_caption", size=(300,97), style=no_caption)
        self.panel=wx.Panel(self)
        self.Butt = wx.Button(self.panel, -1, label="Clear Desktop", pos=(0,0), size=(100,70))
        self.Butt.Bind(wx.EVT_BUTTON, self.Process, self.Butt)
        self.Butt2 = wx.Button(self.panel, -1, label="Undo Clear", pos=(185,0), size=(100,70))
        self.Butt2.Bind(wx.EVT_BUTTON, self.Process2, self.Butt2)
    def Process(self, e):
        PublicDesktop = os.listdir(self.Pdesktop)
        MainDesktop = os.listdir(self.Desktop)
        PublicDesktop.remove('desktop.ini')
        MainDesktop.remove('desktop.ini')

        for items in MainDesktop:
            shutil.move(self.Desktop + items, self.Dfiles)
            print 'Clear'


    def Process2(self, e):
        Dfolder = os.listdir(self.Dfiles)
        Dfolder.remove('desktop.ini')
        for items2 in Dfolder:
            shutil.move(self.Dfiles + items2, self.Desktop)
            print 'undo'





if __name__=='__main__':
    app=wx.App(False)
    frame=Window(parent=None, id=-1)
    frame.Show()
    app.MainLoop()