cola/icons.py

   1 """The only file where icon filenames are mentioned"""
   2 from __future__ import absolute_import, division, unicode_literals
   3 import os
   4
   5 from qtpy import QtGui
   6 from qtpy import QtWidgets
   7
   8 from . import core
   9 from . import qtcompat
  10 from . import resources
  11 from .compat import ustr
  12 from .i18n import N_
  13
  14
  15 KNOWN_FILE_MIME_TYPES = [
  16     ('text', 'file-code.svg'),
  17     ('image', 'file-media.svg'),
  18     ('octet', 'file-binary.svg'),
  19 ]
  20
  21 KNOWN_FILE_EXTENSIONS = {
  22     '.bash': 'file-code.svg',
  23     '.c': 'file-code.svg',
  24     '.cpp': 'file-code.svg',
  25     '.css': 'file-code.svg',
  26     '.cxx': 'file-code.svg',
  27     '.h': 'file-code.svg',
  28     '.hpp': 'file-code.svg',
  29     '.hs': 'file-code.svg',
  30     '.html': 'file-code.svg',
  31     '.java': 'file-code.svg',
  32     '.js': 'file-code.svg',
  33     '.ksh': 'file-code.svg',
  34     '.lisp': 'file-code.svg',
  35     '.perl': 'file-code.svg',
  36     '.pl': 'file-code.svg',
  37     '.py': 'file-code.svg',
  38     '.rb': 'file-code.svg',
  39     '.rs': 'file-code.svg',
  40     '.sh': 'file-code.svg',
  41     '.zsh': 'file-code.svg',
  42 }
  43
  44
  45 def install(themes):
  46     for theme in themes:
  47         icon_dir = resources.icon_dir(theme)
  48         qtcompat.add_search_path('icons', icon_dir)
  49
  50
  51 def icon_themes():
  52     return (
  53         (N_('Default'), 'default'),
  54         (N_('Dark Theme'), 'dark'),
  55         (N_('Light Theme'), 'light'),
  56     )
  57
  58
  59 def name_from_basename(basename):
  60     """Prefix the basename with "icons:" so that git-cola's icons are found
  61
  62     "icons" is registered with Qt's resource system during install().
  63
  64     """
  65     return 'icons:' + basename
  66
  67
  68 def from_name(name):
  69     """Return a QIcon from an absolute filename or "icons:basename.svg" name"""
  70     return QtGui.QIcon(name)
  71
  72
  73 def icon(basename):
  74     """Given a basename returns a QIcon from the corresponding cola icon"""
  75     return from_name(name_from_basename(basename))
  76
  77
  78 def from_theme(name, fallback=None):
  79     """Grab an icon from the current theme with a fallback
  80
  81     Support older versions of Qt checking for fromTheme's availability.
  82
  83     """
  84     if hasattr(QtGui.QIcon, 'fromTheme'):
  85         base, _ = os.path.splitext(name)
  86         if fallback:
  87             qicon = QtGui.QIcon.fromTheme(base, icon(fallback))
  88         else:
  89             qicon = QtGui.QIcon.fromTheme(base)
  90         if not qicon.isNull():
  91             return qicon
  92     return icon(fallback or name)
  93
  94
  95 def basename_from_filename(filename):
  96     """Returns an icon name based on the filename"""
  97     mimetype = core.guess_mimetype(filename)
  98     if mimetype is not None:
  99         mimetype = mimetype.lower()
 100         for filetype, icon_name in KNOWN_FILE_MIME_TYPES:
 101             if filetype in mimetype:
 102                 return icon_name
 103     extension = os.path.splitext(filename)[1]
 104     return KNOWN_FILE_EXTENSIONS.get(extension.lower(), 'file-text.svg')
 105
 106
 107 def from_filename(filename):
 108     basename = basename_from_filename(filename)
 109     return from_name(name_from_basename(basename))
 110
 111
 112 def mkicon(value, default=None):
 113     if value is None and default is not None:
 114         value = default()
 115     elif value and isinstance(value, (str, ustr)):
 116         value = QtGui.QIcon(value)
 117     return value
 118
 119
 120 def from_style(key):
 121     """Maintain a cache of standard icons and return cache entries."""
 122     style = QtWidgets.QApplication.instance().style()
 123     return style.standardIcon(key)
 124
 125
 126 def status(filename, deleted, is_staged, untracked):
 127     if deleted:
 128         icon_name = 'circle-slash-red.svg'
 129     elif is_staged:
 130         icon_name = 'staged.svg'
 131     elif untracked:
 132         icon_name = 'question-plain.svg'
 133     else:
 134         icon_name = basename_from_filename(filename)
 135     return icon_name
 136
 137
 138 # Icons creators and SVG file references
 139
 140 def three_bars():
 141     return icon('three-bars.svg')
 142
 143
 144 def add():
 145     return from_theme('list-add', fallback='plus.svg')
 146
 147
 148 def alphabetical():
 149     return icon('a-z-order.svg')
 150
 151
 152 def branch():
 153     return icon('git-branch.svg')
 154
 155
 156 def check_name():
 157     return name_from_basename('check.svg')
 158
 159
 160 def close():
 161     return icon('x.svg')
 162
 163
 164 def cola():
 165     return icon('git-cola.svg')
 166
 167
 168 def commit():
 169     return icon('document-save-symbolic.svg')
 170
 171
 172 def compare():
 173     return icon('git-compare.svg')
 174
 175
 176 def configure():
 177     return icon('gear.svg')
 178
 179
 180 def copy():
 181     return from_theme('edit-copy.svg')
 182
 183
 184 def default_app():
 185     return icon('telescope.svg')
 186
 187
 188 def dot_name():
 189     return name_from_basename('primitive-dot.svg')
 190
 191
 192 def download():
 193     return icon('file-download.svg')
 194
 195
 196 def discard():
 197     return icon('trashcan.svg')
 198
 199
 200 # folder vs directory: directory is opaque, folder is just an outline
 201 # directory is used for the File Browser, where more contrast with the file
 202 # icons are needed.
 203
 204 def folder():
 205     return icon('folder.svg')
 206
 207
 208 def directory():
 209     return icon('file-directory.svg')
 210
 211
 212 def diff():
 213     return icon('diff.svg')
 214
 215
 216 def edit():
 217     return icon('pencil.svg')
 218
 219
 220 def ellipsis():
 221     return icon('ellipsis.svg')
 222
 223
 224 def external():
 225     return icon('link-external.svg')
 226
 227
 228 def file_code():
 229     return icon('file-code.svg')
 230
 231
 232 def file_text():
 233     return icon('file-text.svg')
 234
 235
 236 def file_zip():
 237     return icon('file-zip.svg')
 238
 239
 240 def fold():
 241     return icon('fold.svg')
 242
 243
 244 def merge():
 245     return icon('git-merge.svg')
 246
 247
 248 def modified():
 249     return icon('modified.svg')
 250
 251
 252 def modified_name():
 253     return name_from_basename('modified.svg')
 254
 255
 256 def new():
 257     return icon('folder-new.svg')
 258
 259
 260 def ok():
 261     return icon('check.svg')
 262
 263
 264 def open_directory():
 265     return icon('folder.svg')
 266
 267
 268 def partial_name():
 269     return name_from_basename('partial.svg')
 270
 271
 272 def pull():
 273     return icon('repo-pull.svg')
 274
 275
 276 def push():
 277     return icon('repo-push.svg')
 278
 279
 280 def question():
 281     return icon('question.svg')
 282
 283
 284 def remove():
 285     return from_theme('list-remove', fallback='circle-slash.svg')
 286
 287
 288 def repo():
 289     return icon('repo.svg')
 290
 291
 292 def reverse_chronological():
 293     return icon('last-first-order.svg')
 294
 295
 296 def save():
 297     return icon('desktop-download.svg')
 298
 299
 300 def search():
 301     return icon('search.svg')
 302
 303
 304 def select_all():
 305     return from_theme('edit-select-all.svg')
 306
 307
 308 def staged():
 309     return icon('staged.svg')
 310
 311
 312 def staged_name():
 313     return name_from_basename('staged.svg')
 314
 315
 316 def star():
 317     return icon('star.svg')
 318
 319
 320 def sync():
 321     return icon('sync.svg')
 322
 323
 324 def tag():
 325     return icon('tag.svg')
 326
 327
 328 def undo():
 329     return from_theme('edit-undo', fallback='edit-undo.svg')
 330
 331
 332 def style_dialog_apply():
 333     return from_style(QtWidgets.QStyle.SP_DialogApplyButton)
 334
 335
 336 def style_dialog_discard():
 337     return from_style(QtWidgets.QStyle.SP_DialogDiscardButton)
 338
 339
 340 def style_dialog_reset():
 341     return from_style(QtWidgets.QStyle.SP_DialogResetButton)
 342
 343
 344 def unfold():
 345     return icon('unfold.svg')
 346
 347
 348 def visualize():
 349     return icon('eye.svg')
 350
 351
 352 def upstream_name():
 353     return name_from_basename('upstream.svg')
 354
 355
 356 def zoom_fit_best():
 357     return icon('zoom-fit-best.svg')
 358
 359
 360 def zoom_in():
 361     return icon('zoom-in.svg')
 362
 363
 364 def zoom_out():
 365     return icon('zoom-out.svg')