1

from __future__ import absolute_import

     2

     3

from django.conf import settings

     4

import django.utils.http

     5

     6

# the top-level pngtile package

     7

from pngtile.store import (

     8

        Error,

     9

        NotFound,

    10

        InvalidImage,

    11

        UncachedImage,

    12

        PNGTileStore,

    13

)

    14

    15

import os.path

    16

    17

class Store (PNGTileStore):

    21

    def __init__ (self, title, tileserver, **opts):

    22

        super(Store, self).__init__(**opts)

    23

    24

        self._title = title

    25

        self.tileserver = tileserver

    26

    27

    def title (self, name):

    28

        if name:

    29

            return os.path.basename(name)

    30

        else:

    31

            return self._title

    32

    33

    def breadcrumb (self, name):

    34

        """

    35

            Yield (name, title) tuples for breadcrumb to name

    36

        """

    37

    38

        path = [ ]

    39

    40

        for part in name.split('/'):

    41

            path.append(part)

    42

    43

            yield '/'.join(path), part

    44

    45

    def tree (self, name):

    46

        """

    47

            Yield [(item, name)] lists for given path.

    48

        """

    49

    50

        # always, starting from root

    51

        parts = []

    52

    53

        if name:

    54

            # also including the dir itself... but not the root twice

    55

            name += '/'

    56

    57

        # walk up the tree from root the dir itself

    58

        for tree_item in name.split('/'):

    59

            tree_name = '/'.join(parts)

    60

    61

            dirs = [ ]

    62

            items = [ ]

    63

    64

            # walk through the items within this tree node

    65

            for item, type in self.list(tree_name):

    66

                if tree_name:

    67

                    item_name = tree_name + '/' + item

    68

                else:

    69

                    item_name = item

    70

    71

                if type:

    72

                    items.append((item, item_name))

    73

                else:

    74

                    dirs.append((item, item_name))

    75

    76

            # tree of parent + item within parent

    77

            yield sorted(dirs), tree_item

    78

    79

            parts.append(tree_item)

    80

    81

    def items (self, name):

    82

        """

    83

            Yield (item, name) for given path.

    84

        """

    85

    86

        for item, type in self.list(name):

    87

            if name:

    88

                item_name = name + '/' + item

    89

            else:

    90

                item_name = item

    91

    92

            if type:

    93

                yield item, item_name

    94

    96

        """

    97

            Return a tileserver URL.

    98

        """

    99

   100

        url = self.tileserver

   101

   102

        if name:

   103

            url += name

   104

   105

        if query:

   106

            url += '?' + django.utils.http.urlencode(query)

   107

   108

        return url

   109

   110

STORE = Store(

   111

    # PNGTileStore

   112

    image_root  = settings.QMSK_PNGTILE_ROOT,

   113

   114

    # Store

   115

    title       = settings.QMSK_PNGTILE_TITLE,

   116

    tileserver  = settings.QMSK_PNGTILE_SERVER,

   117

)