180 lines
5.0 KiB
Python
180 lines
5.0 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2006 Brian Matherly
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
|
|
# $Id:
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Standard python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import os
|
|
from _winreg import *
|
|
from gettext import gettext as _
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GNOME/GTK
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Gramps modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import const
|
|
import _PythonMime
|
|
|
|
|
|
def get_application(type):
|
|
"""Returns the application command and application name of the
|
|
specified mime type"""
|
|
extension = _get_extension(type)
|
|
progId = _get_prog_id(extension)
|
|
|
|
if not progId:
|
|
return None
|
|
|
|
# Find the application associated with this program ID
|
|
aReg = ConnectRegistry(None,HKEY_CLASSES_ROOT)
|
|
subkey = OpenKey(aReg, "%s\shell\open\command" % progId)
|
|
if subkey:
|
|
n,command,type = EnumValue(subkey, 0)
|
|
if type == REG_EXPAND_SZ:
|
|
command = command.replace( '%SystemRoot%',
|
|
os.getenv('SystemRoot') )
|
|
CloseKey(aReg)
|
|
else:
|
|
return None
|
|
CloseKey(subkey)
|
|
|
|
# Find a friendly name for the application
|
|
if command.startswith('"'):
|
|
app = command.split('"')[1]
|
|
elif command.startswith('rundll32.exe'):
|
|
# Get the description of the DLL instead of the application
|
|
app = command.split()[1].split(',')[0]
|
|
else:
|
|
app = command.split()[0]
|
|
|
|
hcu = ConnectRegistry(None,HKEY_CURRENT_USER)
|
|
subkey = OpenKey(hcu, "Software\Microsoft\Windows\ShellNoRoam\MUICache")
|
|
desc = None
|
|
if subkey:
|
|
try:
|
|
desc,ValType = QueryValueEx(subkey, app)
|
|
except:
|
|
# No friendly name exists. Use progId
|
|
desc = progId
|
|
CloseKey(subkey)
|
|
else:
|
|
desc = progId
|
|
CloseKey(hcu)
|
|
|
|
return (command,desc)
|
|
|
|
def get_description(mime_type):
|
|
"""Returns the description of the specfied mime type"""
|
|
desc = None
|
|
extension = _get_extension(mime_type)
|
|
progId = _get_prog_id(extension)
|
|
|
|
if progId:
|
|
aReg = ConnectRegistry(None,HKEY_CLASSES_ROOT)
|
|
desc = QueryValue(aReg, progId)
|
|
CloseKey(aReg)
|
|
|
|
if not desc:
|
|
desc = _("unknown")
|
|
|
|
return desc
|
|
|
|
def get_type(file):
|
|
"""Returns the mime type of the specified file"""
|
|
return _PythonMime.get_type(file)
|
|
|
|
def mime_type_is_defined(mime_type):
|
|
"""
|
|
Return True if a description for a mime type exists.
|
|
"""
|
|
extension = _get_extension(mime_type)
|
|
if extension:
|
|
return True
|
|
else:
|
|
return _PythonMime.mime_type_is_defined(mime_type)
|
|
|
|
_icon_theme = gtk.icon_theme_get_default()
|
|
|
|
def find_mime_type_pixbuf(mime_type):
|
|
return _PythonMime.find_mime_type_pixbuf(mime_type)
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# private functions
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
def _get_extension(mime_type):
|
|
"""
|
|
Return the extension associated with this mime type
|
|
Return None if no association exists
|
|
"""
|
|
extension = None
|
|
try:
|
|
aReg = ConnectRegistry(None,HKEY_CLASSES_ROOT)
|
|
subkey = OpenKey(aReg, "MIME\DataBase\Content Type")
|
|
mimekey = OpenKey(subkey, mime_type)
|
|
extension,type = QueryValueEx(mimekey, "Extension")
|
|
CloseKey(mimekey)
|
|
CloseKey(subkey)
|
|
CloseKey(aReg)
|
|
except:
|
|
extension = None
|
|
|
|
if not extension:
|
|
# Work around for Windows mime problems
|
|
extmap = {
|
|
'application/abiword' : '.abw',
|
|
'application/rtf' : '.rtf',
|
|
}
|
|
if extmap.has_key(mime_type):
|
|
extension = extmap[mime_type]
|
|
|
|
return extension
|
|
|
|
|
|
def _get_prog_id(extension):
|
|
"""
|
|
Return the program ID associated with this extension
|
|
Return None if no association exists
|
|
"""
|
|
if not extension:
|
|
return None
|
|
|
|
try:
|
|
aReg = ConnectRegistry(None,HKEY_CLASSES_ROOT)
|
|
progId = QueryValue(aReg, extension)
|
|
CloseKey(aReg)
|
|
return progId
|
|
except:
|
|
return None
|
|
|