1

#!/usr/bin/env python2.4

     2

#

     3

# DeGAL - A pretty simple web image gallery

     4

# Copyright (C) 2007 Tero Marttila

     5

# http://marttila.de/~terom/degal/

     6

#

     7

# This program is free software; you can redistribute it and/or modify

     8

# it under the terms of the GNU General Public License as published by

     9

# the Free Software Foundation; either version 2 of the License, or

    10

# (at your option) any later version.

    11

#

    12

# This program is distributed in the hope that it will be useful,

    13

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    14

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    15

# GNU General Public License for more details.

    16

#

    17

# You should have received a copy of the GNU General Public License

    18

# along with this program; if not, write to the

    19

# Free Software Foundation, Inc.,

    20

# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

    21

#

    22

    23

import os.path, os

    24

import urllib

    25

import logging

    26

import string

    27

from datetime import datetime

    28

import struct

    29

import base64

    30

import shelve

    31

    32

import PIL

    33

import PIL.Image

    34

    35

import utils

    36

    37

__version__ = '0.2'

    38

    39

TEMPLATE_DIR='templates'

    40

TEMPLATE_EXT='html'

    41

    42

logging.basicConfig(

    43

    level=logging.INFO,

    44

    format="%(name)8s %(levelname)8s   %(lineno)3d %(message)s",

    45

    46

)

    47

    48

tpl = logging.getLogger('tpl')

    49

index = logging.getLogger('index')

    50

prepare = logging.getLogger('prepare')

    51

render = logging.getLogger('render')

    52

    53

tpl.setLevel(logging.WARNING)

    54

    55

#for l in (tpl, index, prepare, render) :

    56

#    l.setLevel(logging.DEBUG)

    57

    58

def readFile (path) :

    59

    fo = open(path, 'r')

    60

    data = fo.read()

    61

    fo.close()

    62

    63

    return data

    64

    65

class Template (object) :

    66

    GLOBALS = dict(

    67

        VERSION=__version__,

    68

    )

    69

    70

    def __init__ (self, name) :

    71

        self.name = name

    72

        self.path = os.path.join(TEMPLATE_DIR, "%s.%s" % (name, TEMPLATE_EXT))

    73

    74

        tpl.debug("Template %s at %s", name, self.path)

    75

    76

        self.content = readFile(self.path)

    77

    78

    def render (self, **vars) :

    79

        content = self.content

    80

    81

        vars.update(self.GLOBALS)

    82

    83

        for name, value in vars.iteritems() :

    84

            content = content.replace('<!-- %s -->' % name, str(value))

    85

    86

        return content

    87

    88

    def renderTo (self, path, **vars) :

    89

        tpl.info("Render %s to %s", self.name, path)

    90

    91

        fo = open(path, 'w')

    92

        fo.write(self.render(**vars))

    93

        fo.close()

    94

    95

gallery_tpl = Template('gallery')

    96

image_tpl = Template('image')

    97

    98

IMAGE_EXTS = ('jpg', 'jpeg', 'png', 'gif', 'bmp')

    99

   100

THUMB_DIR = 'thumbs'

   101

PREVIEW_DIR = 'previews'

   102

TITLE_FILE = 'title.txt'

   103

   104

THUMB_GEOM = (160, 120)

   105

PREVIEW_GEOM = (640, 480)

   106

   107

DEFAULT_TITLE = 'Image gallery'

   108

   112

def isImage (fname) :

   113

    """

   114

        Is the given filename likely to be an image file?

   115

    """

   116

   117

    fname = fname.lower()

   118

    base, ext = os.path.splitext(fname)

   119

    ext = ext.lstrip('.')

   120

   121

    return ext in IMAGE_EXTS

   122

   123

def link (url, title) :

   124

    """

   125

        Returns an <a href=""></a> link tag with the given values

   126

    """

   127

   128

    return "<a href='%s'>%s</a>" % (urllib.quote(url), title)

   129

   130

def dirUp (count=1) :

   131

    """

   132

        Returns a relative path to the directly count levels above the current one

   133

    """

   134

   135

    if not count :

   136

        return '.'

   137

   138

    return os.path.join(*(['..']*count))

   139

   140

def readTitleDescr (path) :

   141

    """

   142

        Read a title.txt or <imgname>.txt file

   143

    """

   144

   145

    if os.path.exists(path) :

   146

        content = readFile(path)

   147

   148

        if '---' in content :

   149

            title, descr = content.split('---', 1)

   150

        else :

   151

            title, descr = content, ''

   152

   153

        return title.strip(), descr.strip()

   154

   155

    return None, None

   156

   157

class Folder (object) :

   158

    def __init__ (self, name='.', parent=None) :

   159

        # the directory name

   160

        self.name = name

   161

   162

        # our parent Folder, or None

   163

        self.parent = parent

   164

   165

        # the path to this dir, as a relative path to the root of the image gallery, always starts with .

   166

        if parent and name :

   167

            self.path = parent.pathFor(name)

   168

        else :

   169

            self.path = name

   170

   171

        # the url-path to the index.html file

   172

        self.html_path = self.path

   173

   174

        # dict of fname -> Folder

   175

        self.subdirs = {}

   176

   177

        # dict of fname -> Image

   178

        self.images = {}

   179

   180

        # our human-friendly title

   181

        self.title = None

   182

   183

        # our long-winded description

   184

        self.descr = ''

   185

   186

        # is this folder non-empty?

   187

        self.alive = None

   188

   189

        # self.images.values(), but sorted by filename

   190

        self.sorted_images = []

   191

   192

        # the ShortURL key to this dir

   193

        self.shorturl_code = None

   194

   195

        # were we filtered out?

   196

        self.filtered = False

   197

   198

    def pathFor (self, *fnames) :

   199

        """

   200

            Return a root-relative path to the given path inside this dir

   201

        """

   202

        return os.path.join(self.path, *fnames)

   203

   204

    def index (self, filters=None) :

   205

        """

   206

            Look for other dirs and images inside this dir. Filters must be either None,

   207

            whereupon all files will be included, or a dict of {filename -> next_filter}.

   208

            If given, only filenames that are present in the dict will be indexed, and in

   209

            the case of dirs, the next_filter will be passed on to that Folder's index

   210

            method.

   211

        """

   212

   213

        index.info("Indexing %s", self.path)

   214

   215

        if filters :

   216

            self.filtered = True

   217

   218

        # iterate through listdir

   219

        for fname in os.listdir(self.path) :

   220

            # the full filesystem path to it

   221

            fpath = self.pathFor(fname)

   222

   223

            # ignore dotfiles

   224

            if fname.startswith('.') :

   225

                index.debug("Skipping dotfile %s", fname)

   226

                continue

   227

   228

            # apply filters

   229

            if filters :

   230

                if fname in filters :

   231

                    next_filter = filters[fname]

   232

                else :

   233

                    index.debug("Skip `%s' as we have a filter", fname)

   234

                    continue

   235

            else :

   236

                next_filter = None

   237

   238

            # recurse into subdirs, but not thumbs/previews

   239

            if os.path.isdir(fpath) and fname not in (THUMB_DIR, PREVIEW_DIR) :

   240

                index.debug("Found subdir %s", fpath)

   241

                f = self.subdirs[fname] = Folder(fname, self)

   242

                f.index(next_filter)   # recursion

   243

   244

            # handle images

   245

            elif os.path.isfile(fpath) and isImage(fname) :

   246

                index.debug("Found image %s", fname)

   247

                self.images[fname] = Image(self, fname)

   248

   249

            # ignore everything else

   250

            else :

   251

                index.debug("Ignoring file %s", fname)

   252

   253

    def prepare (self) :

   254

        """

   255

            Prepare the dir, i.e. sort+prepare the images, as well as recurse

   256

            into subdirs

   257

        """

   258

   259

        prepare.info("Preparing dir %s", self.path)

   260

   261

        # is this folder non-empty?

   262

        alive = False

   263

   264

        # only create thumbs/previews dirs if we have images in here

   265

        if self.images :

   266

            # folder is non-empty

   267

            alive = True

   268

            prepare.info("Have %d images", len(self.images))

   269

   270

            # create the thumb/preview dirs if needed

   271

            for dir in (THUMB_DIR, PREVIEW_DIR) :

   272

                prepare.debug("Checking for existance of %s dir", dir)

   273

                path = self.pathFor(dir)

   274

   275

                if not os.path.isdir(path) :

   276

                    prepare.info("Creating dir %s", path)

   277

                    os.mkdir(path)

   278

   279

            # sort the images

   280

            fnames = self.images.keys()

   281

            fnames.sort()

   282

   283

            prev = None

   284

   285

            # link them together and prepare them

   286

            for fname in fnames :

   287

                img = self.images[fname]

   288

   289

                prepare.debug("Linking %s <-> %s", prev, img)

   290

   291

                img.prev = prev

   292

   293

                if prev :

   294

                    prev.next = img

   295

   296

                prev = img

   297

   298

                img.prepare()

   299

   300

                # add to the sorted images list

   301

                self.sorted_images.append(img)

   302

   303

        # prepare subdirs

   304

        if self.subdirs :

   305

            prepare.info("Have %d subdirs", len(self.subdirs))

   306

   307

            # just recurse into them, we're alive if one of them is

   308

            for dir in self.subdirs.itervalues() :

   309

                if dir.prepare() :

   310

                    alive = True

   311

   312

        # figure out our title

   313

        title_path = self.pathFor(TITLE_FILE)

   314

   315

        title, descr = readTitleDescr(title_path)

   316

   317

        if title :

   318

            prepare.info("Found title/descr")

   319

            self.title = title

   320

            self.descr = descr

   321

            alive = True

   322

   323

        # default title for the root dir

   324

        elif self.name == '.' :

   325

            self.title = 'Index'

   326

   327

        else :

   328

            self.title = self.name

   329

   330

        prepare.debug("Our title is '%s'", self.title)

   331

   332

        # lol ded

   333

        if not alive :

   334

            prepare.info("Dir %s is not alive", self.path)

   335

   336

        self.alive = alive

   337

   338

        return alive

   339

   340

    def getObjInfo (self) :

   341

        """

   342

            Metadata for shorturls2.db

   343

        """

   345

   346

    def linkTag (self) :

   347

        """

   348

            A text-link to this dir

   349

        """

   350

   351

        return link(self.path, self.title)

   352

   353

    def breadcrumb (self) :

   354

        """

   355

            Returns a [(fname, title)] list of this dir's parent dirs

   356

        """

   357

   358

        f = self

   359

        b = []

   360

        d = 0

   361

   362

        while f :

   363

            b.insert(0, (dirUp(d), f.title))

   364

   365

            d += 1

   366

            f = f.parent

   367

   368

        return b

   369

   370

    def inRoot (self, *fnames) :

   371

        """

   372

            Return a relative URL from this dir to the given path in the root dir

   373

        """

   374

   375

        c = len(self.path.split('/')) - 1

   376

   377

        return os.path.join(*((['..']*c) + list(fnames)))

   386

   387

    def render (self) :

   388

        """

   389

            Render the index.html, Images, and recurse into subdirs

   390

        """

   391

   392

        # ded folders are skipped

   393

        if not self.alive :

   394

            render.info("Skipping dir %s", self.path)

   395

            return

   396

   397

        # if this dir's contents were filtered out, then we can't render the index.html, as we aren't aware of all the images in here

   398

        if self.filtered :

   399

            render.warning("Dir `%s' contents were filtered, so we won't render the gallery index again", self.path)

   400

   401

        else :

   402

            # sort the subdirs

   403

            subdirs = self.subdirs.items()

   404

            subdirs.sort()

   405

   406

            # generate the <a href=""></a>'s for the subdirs

   407

            subdir_linkTags = [link(f.name, f.title) for fname, f in subdirs if f.alive]

   408

   409

            # stick them into a list

   410

            if subdir_linkTags :

   411

                directories = "<ul>\n\t<li>%s</li>\n</ul>" % "</li>\n\t<li>".join(subdir_linkTags)

   412

            else :

   413

                directories = ''

   414

   415

            render.info("Rendering %s", self.path)

   421

   468

   469

        # render images

   470

        for img in self.images.itervalues() :

   471

            img.render()

   472

   473

        # recurse into subdirs

   474

        for dir in self.subdirs.itervalues() :

   475

            dir.render()

   476

   477

class Image (object) :

   478

    def __init__ (self, dir, name) :

   479

        # the image filename, e.g. DSC3948.JPG

   480

        self.name = name

   481

   482

        # the Folder object that we are in

   483

        self.dir = dir

   484

   485

        # the relative path from the root to us

   486

        self.path = dir.pathFor(name)

   487

   488

        # the basename+ext, e.g. DSCR3948, .JPG

   489

        self.base_name, self.ext = os.path.splitext(name)

   490

   491

        # the root-relative paths to the thumb and preview images

   492

        self.thumb_path = self.dir.pathFor(THUMB_DIR, self.name)

   493

        self.preview_path = self.dir.pathFor(PREVIEW_DIR, self.name)

   494

   495

        # our user-friendly title

   496

        self.title = name

   497

   498

        # our long-winded description

   499

        self.descr = ''

   500

   501

        # the image before and after us, both may be None

   502

        self.prev = self.next = None

   503

   504

        # the name of the .html gallery view thing for this image, *always* self.name + ".html"

   505

        self.html_name = self.name + ".html"

   506

   507

        # the root-relative path to the gallery view

   508

        self.html_path = self.dir.pathFor(self.html_name)

   509

   510

        #

   511

        # Figured out after prepare

   512

        #

   513

   514

        # (w, h) tuple

   515

        self.img_size = None

   516

   517

        # the ShortURL code for this image

   518

        self.shorturl_code = None

   519

   520

        # what to use in the rendered templates, intended to be overridden by subclasses

   521

        self.series_act = "add"

   522

        self.series_verb = "Add to"

   523

   524

    def prepare (self) :

   525