1

# DeGAL - A pretty simple web image gallery

     2

# Copyright (C) 2007 Tero Marttila

     3

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

     4

#

     5

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

     6

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

     7

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

     8

# (at your option) any later version.

     9

#

    10

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

    11

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

    12

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    13

# GNU General Public License for more details.

    14

#

    15

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

    16

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

    17

# Free Software Foundation, Inc.,

    18

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

    19

#

    20

    21

import struct

    22

import base64

    23

import shelve

    24

import os.path

    25

    26

from log import index

    27

    28

def int2key (id) :

    29

    """

    30

        Turn an integer into a short-as-possible url-safe string

    31

    """

    32

    for type in ('B', 'H', 'I') :

    33

        try :

    34

            return base64.b64encode(struct.pack(type, id), '-_').rstrip('=')

    35

        except struct.error :

    36

            continue

    37

    38

    raise Exception("ID overflow: %s" % id)

    39

    40

def updateDB (root) :

    41

    """

    42

        DeGAL <= 0.2 used a simple key => path mapping, but now we use

    43

        something more structured, key => (type, dirpath, fname), where

    44

    45

        type    - one of 'img', 'dir'

    46

        dirpath - the path to the directory, e.g. '.', './foobar', './foobar/quux'

    47

        fname   - the filename, one of '', 'DSC9839.JPG', 'this.png', etc.

    48

    """

    49

    50

    db = shelve.open('shorturls2', 'c', writeback=True)

    51

    52

    id = db.get('_id', 1)

    53

    54

    dirqueue = [root]

    55

    56

    # dict of path -> obj

    57

    paths = {}

    58

    59

    index.info("Processing ShortURLs...")

    60

    61

    while dirqueue :

    62

        dir = dirqueue.pop(0)

    63

    64

        dirqueue.extend(dir.subdirs.itervalues())

    65

    66

        if dir.alive :

    67

            paths[dir.path] = dir

    68

    69

        for img in dir.images.itervalues() :

    70

            paths[img.path] = img

    71

    72

    for key in db.keys() :

    73

        if key.startswith('_') :

    74

            continue

    75

    76

        type, dirpath, fname = db[key]

    77

    78

        path = os.path.join(dirpath, fname).rstrip(os.sep)

    79

    80

        try :

    81

            paths.pop(path).shorturl_code = key

    82

            index.debug("Code for `%s' is %s", path, key)

    83

    84

        except KeyError :

    85

            index.debug("Path `%s' in DB does not exist?", path)

    86

    87

    for obj in paths.itervalues() :

    88

        key = int2key(id)

    89

        id += 1

    90

    91

        index.info("Alloc code `%s' for `%s'", key, obj.html_path)

    92

    93

        obj.shorturl_code = key

    94

    95

        db[key] = obj.getObjInfo()

    96

    97

    db['_id'] = id

    98

    db.close()