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

    29

    30

def int2key (id) :

    31

    """

    32

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

    33

    """

    34

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

    35

        try :

    36

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

    37

        except struct.error :

    38

            continue

    39

    40

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

    41

    42

def key2int (key) :

    43

    # base64 ignores extra padding, but if it doesn't, it's (4 - len%4), if len%4 != 0

    46

    47

    type = {

    48

        1: 'B',

    49

        2: 'H',

    50

        4: 'I',

    51

    }[len(bytes)]

    52

    53

    return struct.unpack(type, bytes)[0]

    54

    55

class DB (object) :

    56

    def __init__ (self, read_only=True) :

    57

        self.db = shelve.open('shorturls2', read_only and 'r' or 'c')

    58

    59

    def html_path (self, key, index) :

    60

        type, dirpath, fname = self.db[key]

    61

    62

        if type == 'img' :

    63

            fname += '.html'

    64

        elif type == 'dir' :

    65

            fname = ''

    66

    67

        if index :

    68

            dirpath = '../%s' % dirpath

    69

    70

            if type == 'dir' and index > 1 : 

    71

                fname = 'index_%s.html' % (index - 1)

    72

    73

        return os.path.join(dirpath, fname)

    74

    75

    def image_info (self, key) :

    76

        type, dirpath, fname = self.db[key]

    77

    78

        if type != 'img' :

    79

            raise ValueError("%s is not an img" % key)

    80

    81

        return dirpath, fname

    82

    83

    def shorturls_for (self, paths) :

    84

        ret = []

    85

    86

        for key in self.db.keys() :

    87

            if key.startswith('_') :

    88

                continue

    89

    90

            type, dir, fname = self.db[key]

    91

            path = os.path.join(dir.lstrip('.').lstrip('/'), fname) 

    92

            if path in paths :

    93

                ret.append(key)

    94

                paths.remove(path)

    95

    96

        if paths :

    97

            raise ValueError("Paths not found: %s" % " ".join(paths))

    98

    99

        return ret

   100

   101

def html_path (key, index=None) :

   103

   117

   118

def image_info (key) :

   119

    res = db.select("""SELECT dirpath, filename FROM images WHERE id=?""", key2int(key)).fetchone()

   120

   121

    if res :

   122

        return res

   123

   124

    else :

   125

        raise KeyError(key)

   126

   127

def get_images (keys) :

   128

    res = [db.select("""SELECT dirpath, filename FROM images WHERE id=?""", key2int(key)).fetchone() for key in keys]

   129

   130

    # don't mind if we don't get as many as we asked for?

   131

    if res :

   132

        return res

   133

   134

    else :

   135

        raise KeyError(keys)

   136