profhandler.py

   1 # -*- coding: utf-8 -*-
   2
   3 """
   4 Dboxswitch dropbox profile switcher
   5
   6 license: Modified BSD License
   7
   8 Copyright (c) 2012,  <stack@inventati.org>
   9 All rights reserved.
  10
  11 Redistribution and use in source and binary forms, with or without
  12 modification, are permitted provided that the following conditions are met:
  13     * Redistributions of source code must retain the above copyright
  14       notice, this list of conditions and the following disclaimer.
  15     * Redistributions in binary form must reproduce the above copyright
  16       notice, this list of conditions and the following disclaimer in the
  17       documentation and/or other materials provided with the distribution.
  18     * Neither the name of the <organization> nor the
  19       names of its contributors may be used to endorse or promote products
  20       derived from this software without specific prior written permission.
  21
  22 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  23 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  24 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  25 DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
  26 DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  27 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  29 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  31 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32
  33 """
  34 import platform
  35 import re
  36 import shutil
  37 import os
  38 import errno
  39 import signal
  40
  41 from apperror import AppError
  42 from settings import appconf
  43
  44 class ProfHandler():
  45
  46     def __init__(self):
  47
  48         #create profile directory if not exists
  49         try:
  50             os.makedirs(self.getProfileFolder())
  51         except OSError, e:
  52             if e.errno != errno.EEXIST:
  53                 raise
  54
  55         #compile regular expression for validating profile names
  56         self.reg = re.compile("[a-zA-Z0-9_-]+")
  57
  58         #patch symlink on windows
  59         if platform.system() is 'Windows':
  60             os.symlink = winsymlink
  61
  62     def getProfilesList(self):
  63         """ Generate and returns the profiles
  64             it assumes that self.pdir is defined """
  65         #this is generated every time to handle the case of the user renaming the directories by hand
  66         return [os.path.join(self.pdir, f) for f in os.listdir(self.pdir)]
  67
  68     def getProfileFolder(self):
  69         """ Generates, in a os dependant way, the local folder where all profiles are stored """
  70         try:
  71             #directory path is cached
  72             return self.pdir
  73         except AttributeError:
  74             pl = platform.system()
  75             if pl == "Linux":
  76                 try:
  77                     from xdg.BaseDirectory import xdf_data_home
  78                     self.pdir = os.path.join(xdg_data_home, appconf.appname)
  79                 except:
  80                     self.pdir = os.path.join(os.path.expanduser('~'),".local/share",appconf.appname)
  81             elif pl == 'Windows':
  82                 self.pdir = os.path.join(os.getenv("APPDATA"), appconf.appname)
  83             elif pl == 'Darwin':
  84                 self.pdir =  os.path.join(os.path.expanduser('~'),"."+appconf.appname)
  85             elif pl == None:
  86                 raise AppError('Operative system NOT supported.')
  87
  88             return self.pdir
  89
  90     def addProfile(self, profileName):
  91         """ Create a profile """
  92
  93         print("Creating a new profile")
  94         if self.isValidProfileName(profileName):
  95             os.makedirs(os.path.join(self.getProfileFolder(), profileName))
  96         else:
  97             raise AppError('Profile Name not valid.\nAllowed only ascii characters.')
  98         print("Profile "+profileName+" created.")
  99
 100     def delProfile(self, profileName):
 101         """ Delete a profile """
 102
 103         print("Deleting profile")
 104         if self.isValidProfileName(profileName):
 105             try:
 106                 #recursively delete the profile directory
 107                 shutil.rmtree(os.path.join(self.pdir, profileName))
 108             except:
 109                 raise AppError('Profile Name does not exists')
 110         else:
 111             raise AppError('Profile Name not valid')
 112         print("Profile "+profileName+" created.")
 113
 114     def isCurrentProfile(self, ppath):
 115         """ Returns true if the current profile path is currently activated """
 116
 117         pl = platform.system()
 118         if pl in ('Linux','Darwin'):
 119             if os.path.exists(self.getDropboxDirectory()):
 120                 return True if os.readlink(self.getDropboxDirectory()) == ppath else False
 121             else:
 122                 return False
 123
 124     def isValidProfileName(self, pname):
 125
 126         if self.reg.match(pname) is not None:
 127             return True
 128         else:
 129             return False
 130
 131     def activateProfile(self, ppath):
 132         pl = platform.system()
 133         if ppath in self.getProfilesList():
 134             self.stopDropbox()
 135             try:
 136                 if pl in ('Linux','Darwin'):
 137                     if os.path.exists(self.getDropboxDirectory()):
 138                         os.unlink(self.getDropboxDirectory())
 139                     os.symlink(ppath, self.getDropboxDirectory())
 140                 else:
 141                     raise NotImplementedError, "Not implemented yet."
 142             except IOError as e:
 143                 raise AppError('Error on activating Profile: '+ self.getBaseProfileName(ppath))
 144             self.startDropbox()
 145         else:
 146             raise AppError("Trying to acrivate non existant profile")
 147
 148     def getBaseProfileName(self, ppath):
 149         """ Returns the base name given a profile returned by getProfilesList """
 150
 151         return os.path.basename(ppath)
 152
 153     def getDropboxDirectory(self):
 154         pl = platform.system()
 155         if pl in ('Linux', 'Darwin'):
 156             return os.path.join(os.path.expanduser('~'),".dropbox")
 157         elif pl == 'Windows':
 158             assert os.environ.has_key('APPDATA'), Exception('APPDATA env variable not found')
 159             return os.path.join(os.environ['APPDATA'],'Dropbox')
 160         else:
 161             raise NotImplementedError, "Not implemented yet."
 162
 163     def stopDropbox(self):
 164         """ Stop dropbox Daemon """
 165         pl = platform.system()
 166         if pl == 'Linux':
 167             os.system("dropbox stop")
 168         if pl in ('Linux','Darwin'):
 169             pidfile = os.path.expanduser("~/.dropbox/dropbox.pid")
 170             try:
 171                 with open(pidfile, "r") as f:
 172                     pid = int(f.read())
 173                     os.kill(pid, signal.SIGTERM)
 174             except:
 175                 pass
 176
 177     def startDropbox(self):
 178         """ Sart dropbox Daemon """
 179
 180         pl = platform.system()
 181         if pl == 'Linux':
 182             try:
 183                 os.system("dropbox start -i")
 184             except:
 185                 raise AppError(u"Could not start dropbox.")
 186         elif pl == 'Darwin':
 187             os.system("/Applications/Dropbox.app/Contents/MacOS/Dropbox &")
 188
 189
 190 __CSL = None
 191 def winsymlink(source, link_name):
 192     '''symlink(source, link_name)
 193        Creates a symbolic link pointing to source named link_name.
 194         Used to patch the nonexistant version on windows for python 2.6'''
 195     global __CSL
 196     if __CSL is None:
 197         import ctypes
 198         csl = ctypes.windll.kernel32.CreateSymbolicLinkW
 199         csl.argtypes = (ctypes.c_wchar_p, ctypes.c_wchar_p, ctypes.c_uint32)
 200         csl.restype = ctypes.c_ubyte
 201         __CSL = csl
 202     flags = 0
 203     if source is not None and os.path.isdir(source):
 204         flags = 1
 205     if __CSL(link_name, source, flags) == 0:
 206         raise ctypes.WinError()
 207