1

"""

     2

    Image handling.

     3

"""

     4

     5

     6

from werkzeug import Request, Response, exceptions

     8

     9

import pngtile.tile

    11

import pypngtile

    12

    13

import json

    14

import os, os.path

    15

    16

def dir_list (root):

    17

    """

    18

        Yield a series of directory items to show for the given dir

    19

    """

    20

    21

    for name in os.listdir(root):

    22

        path = os.path.join(root, name)

    23

    24

        # skip dotfiles

    25

        if name.startswith('.'):

    26

            continue

    27

    28

        # show dirs

    29

        if os.path.isdir(path):

    31

    32

        # examine ext

    33

        if '.' in name:

    34

            name_base, name_type = name.rsplit('.', 1)

    35

        else:

    36

            name_base = name

    37

            name_type = None

    38

    39

        # show .png files with a .cache file

    40

        if name_type in ImageApplication.IMAGE_TYPES and os.path.exists(os.path.join(root, name_base + '.cache')):

    41

            yield name

    42

    43

class ImageApplication (BaseApplication):

    44

    45

    STYLESHEETS = (

    46

        'https://maxcdn.bootstrapcdn.com/bootstrap/3.2.0/css/bootstrap.min.css',

    47

        'https://maxcdn.bootstrapcdn.com/bootstrap/3.2.0/css/bootstrap-theme.min.css',

    48

        'http://cdn.leafletjs.com/leaflet-0.7.3/leaflet.css',

    49

        '/static/pngtile/map.css',

    50

    )

    51

    52

    SCRIPTS = (

    53

        'https://ajax.googleapis.com/ajax/libs/jquery/1.11.1/jquery.min.js',

    54

        'https://maxcdn.bootstrapcdn.com/bootstrap/3.2.0/js/bootstrap.min.js',

    55

        'http://cdn.leafletjs.com/leaflet-0.7.3/leaflet.js',

    56

        '/static/pngtile/map.js',

    57

    )

    58

    64

        BaseApplication.__init__(self, **opts)

    67

    68

    def render_dir_breadcrumb (self, name):

    69

        path = []

    70

    71

        yield html.li(html.a(href='/', *[u"Index"]))

    72

    73

        if name:

    74

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

    75

                path.append(part)

    76

    78

    79

    def render_dir (self, request, name, items):

    80

        """

    81

            request:    werkzeug.Request

    82

            name:       /.../... url to dir

    83

            items:      [...] items in dir

    84

        """

    85

    86

        if name:

    87

            title = name

    88

        else:

    89

            title = "Index"

    90

    91

        return self.render_html(

    92

            title       = name,

    93

            body        = (

    94

                html.div(class_='container', *[

    95

                    html.h1(title),

    96

                    html.div(*[

    97

                        html.ol(class_='breadcrumb', *self.render_dir_breadcrumb(name)),

    98

                    ]),

    99

                    html.div(class_='list', *[

   100

                        html.ul(class_='list-group', *[html.li(class_='list-group-item', *[

   102

                            ]) for item in items]

   103

                        ),

   104

                    ]),

   105

                ]),

   106

            ),

   107

        )

   108

   109

    def render_image (self, request, image, name):

   110

        """

   111

            request:    werkzeug.Request

   112

            image:      pypngtile.Image

   114

        """

   115

   116

        image_info = image.info()

   117

   118

        map_config = dict(

   121

            tile_name       = name,

   122

   123

            tile_size       = pngtile.tile.TILE_SIZE,

   124

            tile_zoom       = pngtile.tile.MAX_ZOOM,

   125

   128

            image_name      = name,

   129

            image_width     = image_info['img_width'],

   130

            image_height    = image_info['img_height'],

   131

        )

   132

   133

        return self.render_html(

   134

            title       = name,

   135

            body        = (

   136

                html.div(id='wrapper', *[

   137

                    html.div(id='map')

   138

                ]),

   139

            ),

   140

            end         = (

   141

                html.script("""$(function() {{ map_init({map_config}); }});""".format(map_config=json.dumps(map_config))),

   142

            ),

   143

        )

   144

   145

    def handle_dir (self, request, name, path):

   146

        """

   147

            Generate response for directory listing.

   148

        """

   149

   154

        items = sorted(dir_list(path))

   155

   156

        html = self.render_dir(request, name, items)

   157

   158

        return Response(html, content_type="text/html")

   159

   160

    def handle_image (self, request, name, path):

   161

        """

   162

            Generate Response for image request.

   163

        """

   175

        try:

   176

            image, name = self.get_image(request.path)

   177

        except pypngtile.Error as error:

   178

            raise exceptions.BadRequest(str(error))

   179

   180

        html = self.render_image(request, image, name)

   181

   182

        return Response(html, content_type="text/html")

   183

   184

    def handle (self, request):

   185

        """

   186

            Handle request for an image

   187

        """

   188

   189

        name, path = self.lookup_path(request.path)

   190

   191

        # determine type

   192

        if '/' in name:

   193

            _, name_base = name.rsplit('/', 1)

   194

        else:

   195

            name_base = name

   196

   197

        if '.' in name_base:

   198

            name_base, name_type = name_base.rsplit('.', 1)

   199

        else:

   200

            name_type = None

   201

   202

        # determine handler

   203

        if os.path.isdir(path):

   204

            return self.handle_dir(request, name, path)

   205

   206

        elif name_type and name_type in self.IMAGE_TYPES:

   207

            return self.handle_image(request, name, path)

   208

   209

        else:

   210

            raise exceptions.NotFound(path)

   211