1

import os, os.path

     2

import pypngtile as pt

     3

     4

from werkzeug import Response, exceptions

     5

     6

from pngtile import render

     7

     8

# path to images

     9

DATA_ROOT = os.environ.get("PNGTILE_DATA_PATH") or os.path.abspath('data/')

    10

    11

# only open each image once

    12

IMAGE_CACHE = {}

    13

    14

### Parse request data

    15

def get_path (req_path) :

    16

    """

    17

        Returns the name and path requested

    18

    """

    19

    20

    # check DATA_ROOT exists..

    21

    if not os.path.isdir(DATA_ROOT) :

    22

        raise exceptions.InternalServerError("Missing DATA_ROOT")

    23

    24

    # path to image

    25

    image_name = req_path.lstrip('/')

    26

    27

    # build absolute path

    28

    image_path = os.path.abspath(os.path.join(DATA_ROOT, image_name))

    29

    30

    # ensure the path points inside the data root

    31

    if not image_path.startswith(DATA_ROOT) :

    32

        raise exceptions.NotFound(image_name)

    33

    34

    return image_name, image_path

    35

    36

def get_image (name, path) :

    37

    """

    38

        Gets an Image object from the cache, ensuring that the cached is available

    39

    """

    40

    41

    # get Image object

    42

    if path in IMAGE_CACHE :

    43

        # get from cache

    44

        image = IMAGE_CACHE[path]

    45

    46

    else :

    47

        # open

    48

        image = pt.Image(path)

    49

    50

        # check

    51

        if image.status() not in (pt.CACHE_FRESH, pt.CACHE_STALE) :

    52

            raise exceptions.InternalServerError("Image cache not available: %s" % name)

    53

    54

        # load

    55

        image.open()

    56

    57

        # cache

    58

        IMAGE_CACHE[path] = image

    59

    60

    return image

    61

    62

    63

### Handle werkzeug.Request objects -> werkzeug.Response

    64

def handle_dir (req, name, path) :

    65

    """

    66

        Handle request for a directory

    67

    """

    68

    69

    prefix = os.path.dirname(req.script_root).rstrip('/')

    70

    71

    return Response(render.dir_html(prefix, name, path), content_type="text/html")

    72

    73

    74

    75

def handle_img_viewport (req, image, name) :

    76

    """

    77

        Handle request for image viewport

    78

    """

    79

    80

    prefix = os.path.dirname(req.script_root).rstrip('/')

    81

    82

    # viewport

    83

    return Response(render.img_html(prefix, name, image), content_type="text/html")

    84

    85

    86

def handle_img_region (req, image, cache) :

    87

    """

    88

        Handle request for an image region

    89

    """

    90

    91

    # specific image

    92

    width = int(req.args['w'])

    93

    height = int(req.args['h'])

    94

    cx = int(req.args['cx'])

    95

    cy = int(req.args['cy'])

    96

    zoom = int(req.args.get('zl', "0"))

    97

    98

    try :

    99

        # yay full render

   100

        return Response(render.img_png_region(image, cx, cy, zoom, width, height, cache), content_type="image/png")

   101

   102

    except ValueError, ex :

   103

        # too large

   104

        raise exceptions.Forbidden(str(ex))

   105

   106

   107

def handle_img_tile (req, image, cache) :

   108

    """

   109

        Handle request for image tile

   110

    """

   111

   112

    # tile

   113

    x = int(req.args['x'])

   114

    y = int(req.args['y'])

   115

    zoom = int(req.args.get('zl', "0"))

   116

   117

    # cache?

   118

   119

    # yay render

   120

    return Response(render.img_png_tile(image, x, y, zoom, cache), content_type="image/png")

   121

   122

## Dispatch req to handle_img_*

   123

def handle_img (req, name, path, cache) :

   124

    """

   125

        Handle request for an image

   126

    """

   127

   128

    # get image object

   129

    image = get_image(name, path)

   130

   131

    # what view?

   132

    if not req.args :

   133

        return handle_img_viewport(req, image, name)

   134

   135

    elif 'w' in req.args and 'h' in req.args and 'cx' in req.args and 'cy' in req.args :

   136

        return handle_img_region(req, image, cache)

   137

   138

    elif 'x' in req.args and 'y' in req.args :

   139

        return handle_img_tile(req, image, cache)

   140

   141

    else :

   142

        raise exceptions.BadRequest("Unknown args")

   143

   144

   145

   146

## Dispatch request to handle_*

   147

def handle_req (req, cache) :

   148

    """

   149

        Main request handler

   150

    """

   151

   152

    # decode req

   153

    name, path = get_path(req.path)

   154

   155

    # determine dir/image

   156

    if os.path.isdir(path) :

   157

        # directory

   158

        return handle_dir(req, name, path)

   159

   160

    elif not os.path.exists(path) :

   161

        # no such file

   162

        raise exceptions.NotFound(name)

   163

   164

    elif not name or not name.endswith('.png') :

   165

        # invalid file

   166

        raise exceptions.BadRequest("Not a PNG file")

   167

   168

    else :

   169

        # image

   170

        return handle_img(req, name, path, cache)

   171

   172