+# -*- coding: utf-8 -*-
+
+"""
+Dboxswitch dropbox profile switcher
+
+license: Modified BSD License
+
+Copyright (c) 2012, <stack@inventati.org>
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ * Neither the name of the <organization> nor the
+ names of its contributors may be used to endorse or promote products
+ derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
+DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+"""
+import platform
+import re
+import shutil
+import os
+import errno
+import signal
+
+from apperror import AppError
+from settings import appconf
+
+class ProfHandler():
+
+ def __init__(self):
+
+ #create profile directory if not exists
+ try:
+ os.makedirs(self.getProfileFolder())
+ except OSError, e:
+ if e.errno != errno.EEXIST:
+ raise
+
+ #compile regular expression for validating profile names
+ self.reg = re.compile("[a-zA-Z0-9_-]+")
+
+ #patch symlink on windows
+ if platform.system() is 'Windows':
+ os.symlink = winsymlink
+
+ def getProfilesList(self):
+ """ Generate and returns the profiles
+ it assumes that self.pdir is defined """
+ #this is generated every time to handle the case of the user renaming the directories by hand
+ return sorted([os.path.join(self.pdir, f) for f in os.listdir(self.pdir)])
+
+ def getProfileFolder(self):
+ """ Generates, in a os dependant way, the local folder where all profiles are stored """
+ try:
+ #directory path is cached
+ return self.pdir
+ except AttributeError:
+ pl = platform.system()
+ if pl == "Linux":
+ try:
+ from xdg.BaseDirectory import xdf_data_home
+ self.pdir = os.path.join(xdg_data_home, appconf.appname)
+ except:
+ self.pdir = os.path.join(os.path.expanduser('~'),".local/share",appconf.appname)
+ elif pl == 'Windows':
+ self.pdir = os.path.join(os.getenv("APPDATA"), appconf.appname)
+ elif pl == 'Darwin':
+ self.pdir = os.path.join(os.path.expanduser('~'),"."+appconf.appname)
+ elif pl == None:
+ raise AppError('Operative system NOT supported.')
+
+ return self.pdir
+
+ def addProfile(self, profileName):
+ """ Create a profile """
+
+ print("Creating a new profile")
+ if self.isValidProfileName(profileName):
+ try:
+ os.makedirs(os.path.join(self.getProfileFolder(), profileName))
+ except OSError,e:
+ if e.errno == errno.EEXIST:
+ raise AppError("Profile exists.")
+ else:
+ raise AppError(str(e))
+ else:
+ raise AppError('Profile Name not valid.\nAllowed only ascii characters.')
+ print("Profile "+profileName+" created.")
+
+ def delProfile(self, profileName):
+ """ Delete a profile """
+
+ print("Deleting profile")
+ if self.isValidProfileName(profileName):
+ try:
+ #recursively delete the profile directory
+ shutil.rmtree(os.path.join(self.pdir, profileName))
+ except:
+ raise AppError('Profile Name does not exists')
+ else:
+ raise AppError('Profile Name not valid')
+ print("Profile "+profileName+" deleted.")
+
+ def isCurrentProfile(self, ppath):
+ """ Returns true if the current profile path is currently activated """
+
+ pl = platform.system()
+ if pl in ('Linux','Darwin'):
+ if os.path.exists(self.getDropboxDirectory()):
+ return True if os.readlink(self.getDropboxDirectory()) == ppath else False
+ else:
+ return False
+
+ def isValidProfileName(self, pname):
+
+ if self.reg.match(pname) is not None:
+ return True
+ else:
+ return False
+
+ def activateProfile(self, ppath):
+ pl = platform.system()
+ if ppath in self.getProfilesList():
+ self.stopDropbox()
+ try:
+ if pl in ('Linux','Darwin'):
+ if os.path.exists(self.getDropboxDirectory()):
+ os.unlink(self.getDropboxDirectory())
+ os.symlink(ppath, self.getDropboxDirectory())
+ else:
+ raise NotImplementedError, "Not implemented yet."
+ except IOError as e:
+ raise AppError('Error on activating Profile: '+ self.getBaseProfileName(ppath))
+ self.startDropbox()
+ else:
+ raise AppError("Trying to acrivate non existant profile")
+
+ def getBaseProfileName(self, ppath):
+ """ Returns the base name given a profile returned by getProfilesList """
+
+ return os.path.basename(ppath)
+
+ def getDropboxDirectory(self):
+ pl = platform.system()
+ if pl in ('Linux', 'Darwin'):
+ return os.path.join(os.path.expanduser('~'),".dropbox")
+ elif pl == 'Windows':
+ assert os.environ.has_key('APPDATA'), Exception('APPDATA env variable not found')
+ return os.path.join(os.environ['APPDATA'],'Dropbox')
+ else:
+ raise NotImplementedError, "Not implemented yet."
+
+ def stopDropbox(self):
+ """ Stop dropbox Daemon """
+ pl = platform.system()
+ if pl == 'Linux':
+ os.system("dropbox stop")
+ if pl in ('Linux','Darwin'):
+ pidfile = os.path.expanduser("~/.dropbox/dropbox.pid")
+ try:
+ with open(pidfile, "r") as f:
+ pid = int(f.read())
+ os.kill(pid, signal.SIGTERM)
+ except:
+ pass
+
+ def startDropbox(self):
+ """ Sart dropbox Daemon """
+
+ pl = platform.system()
+ if pl == 'Linux':
+ try:
+ os.system("dropbox start -i")
+ except:
+ raise AppError(u"Could not start dropbox.")
+ elif pl == 'Darwin':
+ os.system("/Applications/Dropbox.app/Contents/MacOS/Dropbox &")
+
+
+__CSL = None
+def winsymlink(source, link_name):
+ '''symlink(source, link_name)
+ Creates a symbolic link pointing to source named link_name.
+ Used to patch the nonexistant version on windows for python 2.6'''
+ global __CSL
+ if __CSL is None:
+ import ctypes
+ csl = ctypes.windll.kernel32.CreateSymbolicLinkW
+ csl.argtypes = (ctypes.c_wchar_p, ctypes.c_wchar_p, ctypes.c_uint32)
+ csl.restype = ctypes.c_ubyte
+ __CSL = csl
+ flags = 0
+ if source is not None and os.path.isdir(source):
+ flags = 1
+ if __CSL(link_name, source, flags) == 0:
+ raise ctypes.WinError()
+
v.licheni.net / stack / code / dboxswitch.git / blobdiff