def create_default_style(layer_id, style_name, style_type, geom_type, count): layer = Layer.objects.get(id=int(layer_id)) minscaledenominator = -1 maxscaledenominator = -1 if count and count > 200000: minscaledenominator = 0 maxscaledenominator = 50000 style = Style(name=style_name, title=_('Default style for: ') + layer.title, is_default=True, type=style_type, minscale=minscaledenominator, maxscale=maxscaledenominator) style.save() style_layer = StyleLayer(style=style, layer=layer) style_layer.save() if style.is_default: layer_styles = StyleLayer.objects.filter(layer=layer) for ls in layer_styles: s = Style.objects.get(id=ls.style.id) s.is_default = False symbol_type = None if geom.isPoint(geom_type): symbol_type = 'MarkSymbolizer' elif geom.isLine(geom_type): symbol_type = 'LineSymbolizer' elif geom.isPolygon(geom_type): symbol_type = 'PolygonSymbolizer' elif geom.isRaster(geom_type): symbol_type = 'RasterSymbolizer' rule = Rule(style=style, name='Default symbol', title=_('Default symbol'), abstract='', filter=str(""), minscale=minscaledenominator, maxscale=maxscaledenominator, order=0) rule.save() if symbol_type == 'PolygonSymbolizer': symbolizer = PolygonSymbolizer(rule=rule, order=0, fill='#383838', fill_opacity=0.6, stroke='#000000', stroke_width=1, stroke_opacity=1.0, stroke_dash_array='none') symbolizer.save() elif symbol_type == 'LineSymbolizer': symbolizer = LineSymbolizer(rule=rule, order=0, stroke='#000000', stroke_width=1, stroke_opacity=1.0, stroke_dash_array='none') symbolizer.save() elif symbol_type == 'MarkSymbolizer': symbolizer = MarkSymbolizer(rule=rule, order=0, opacity=1.0, size=8, rotation=0, well_known_name='circle', fill='#383838', fill_opacity=0.6, stroke='#000000', stroke_width=1, stroke_opacity=1.0, stroke_dash_array='none') symbolizer.save() elif symbol_type == 'RasterSymbolizer': color_map = ColorMap(type='ramp', extended=False) color_map.save() symbolizer = RasterSymbolizer(rule=rule, color_map=color_map, order=0, opacity=1.0) symbolizer.save() sld_body = sld_builder.build_sld(layer, style) return sld_body
def create_style(request, json_data, layer_id, is_preview=False, has_custom_legend=None): layer = Layer.objects.get(id=int(layer_id)) datastore = layer.datastore workspace = datastore.workspace gs = geographic_servers.get_instance().get_server_by_id( workspace.server.id) layer_styles = StyleLayer.objects.filter(layer=layer) is_default = False if not is_preview: is_default = json_data.get('is_default') if is_default: for ls in layer_styles: s = Style.objects.get(id=ls.style.id) s.is_default = False s.save() else: has_default_style = False for ls in layer_styles: s = Style.objects.get(id=ls.style.id) if s.is_default: has_default_style = True if not has_default_style: is_default = True name = json_data.get('name') if is_preview: name = name + '__tmp' is_default = False style = Style(name=name, title=json_data.get('title'), is_default=is_default, type='CT') if has_custom_legend == 'true': style.has_custom_legend = True legend_name = 'legend_' + ''.join( random.choice(string.ascii_uppercase) for i in range(6)) + '.png' legend_path = utils.check_custom_legend_path() style.custom_legend_url = utils.save_custom_legend( legend_path, request.FILES['file'], legend_name) else: style.has_custom_legend = False style.save() style_layer = StyleLayer(style=style, layer=layer) style_layer.save() json_rule = json_data.get('rule') filter_text = "" if json_rule.get('filter').__len__() != 0: filter_text = str(json.dumps(json_rule.get('filter'))) rule = Rule(style=style, name=json_rule.get('name'), title=json_rule.get('title'), abstract="", filter=filter_text, minscale=json_rule.get('minscale'), maxscale=json_rule.get('maxscale'), order=json_rule.get('order')) rule.save() color_map = ColorMap(type='ramp', extended=False) color_map.save() symbolizer = RasterSymbolizer(rule=rule, color_map=color_map, order=0, opacity=1.0) symbolizer.save() color_map_entry_list = [] for entry in json_data.get('entries'): json_entry = json.loads(entry.get('json')) color_map_entry = ColorMapEntry(color_map=color_map, order=int(json_entry.get('order')), color=json_entry.get('color'), quantity=json_entry.get('quantity'), label=json_entry.get('label'), opacity=json_entry.get('opacity')) color_map_entry_list.append(color_map_entry) order = 0 color_map_entry_list = sorted( color_map_entry_list, key=lambda color_map_entry: float(color_map_entry.quantity)) for cme in color_map_entry_list: cme.order = order cme.save() order = order + 1 sld_body = sld_builder.build_sld(layer, style) if is_preview: if gs.createOverwrittenStyle(style.name, sld_body, True): return True else: return False else: if gs.createOverwrittenStyle(style.name, sld_body, True): if not is_preview: gs.setLayerStyle(layer, style.name, style.is_default) return True else: return False
def update_style(request, json_data, layer, gs, style, is_preview=False, has_custom_legend=None): is_default = json_data.get('is_default', False) is_default = utils.set_default_style(layer, gs, style=style, is_preview=is_preview, is_default=is_default) if has_custom_legend == 'true': style.has_custom_legend = True legend_name = 'legend_' + ''.join( random.choice(string.ascii_uppercase) for i in range(6)) + '.png' legend_path = utils.check_custom_legend_path() if 'file' in request.FILES: style.custom_legend_url = utils.save_custom_legend( legend_path, request.FILES['file'], legend_name) else: style.has_custom_legend = False style.title = json_data.get('title') if json_data.get('minscale') != '': style.minscale = json_data.get('minscale') else: style.minscale = -1 if json_data.get('maxscale') != '': style.maxscale = json_data.get('maxscale') else: style.maxscale = -1 style.is_default = is_default style.save() json_rule = json_data.get('rule') filter_text = "" if json_rule.get('filter').__len__() != 0: filter_text = str(json.dumps(json_rule.get('filter'))) if json_data.get('minscale') != '': minscale = json_rule.get('minscale') else: minscale = -1 if json_data.get('maxscale') != '': maxscale = json_rule.get('maxscale') else: maxscale = -1 rule_old = Rule.objects.get(style=style) rule_old.delete() rule = Rule(style=style, name=json_rule.get('name'), title=json_rule.get('title'), abstract='', filter=filter_text, minscale=minscale, maxscale=maxscale, order=json_rule.get('order')) rule.save() for s in Symbolizer.objects.filter(rule=rule): s.rastersymbolizer.color_map.delete() s.delete() color_map = ColorMap(type='ramp', extended=False) color_map.save() symbolizer = RasterSymbolizer(rule=rule, color_map=color_map, order=0, opacity=1.0) symbolizer.save() color_map_entry_list = [] for entry in json_data.get('entries'): json_entry = json.loads(entry.get('json')) color_map_entry = ColorMapEntry(color_map=color_map, order=int(json_entry.get('order')), color=json_entry.get('color'), quantity=json_entry.get('quantity'), label=json_entry.get('label'), opacity=json_entry.get('opacity')) color_map_entry_list.append(color_map_entry) order = 0 color_map_entry_list = sorted( color_map_entry_list, key=lambda color_map_entry: float(color_map_entry.quantity)) for cme in color_map_entry_list: cme.order = order cme.save() order = order + 1 sld_body = sld_builder.build_sld(layer, style) if is_preview: if gs.createOverwrittenStyle(style.name, sld_body, True): return style else: if gs.updateStyle(layer, style.name, sld_body): gs.setLayerStyle(layer, style.name, style.is_default) return style