From 5cea5207c4916e56e9b6dc1dce7f66ed54fcd597 Mon Sep 17 00:00:00 2001 From: portnov Date: Tue, 10 Jun 2008 18:54:06 +0600 Subject: [PATCH] Add new modules. Add head module (hms.py). --- hms.py | 6 ++++ modules.py | 28 +++++++++++------ modules/blocks.py | 6 ++- modules/bug_chart.py | 74 ++++++++++++++++++++++++++++++++++++++++++++++ modules/last_comments.py | 27 +++++++++++++++++ modules/login.py | 17 ++++++++++ modules/menu.py | 14 +++++++++ 7 files changed, 160 insertions(+), 12 deletions(-) create mode 100644 hms.py create mode 100755 modules/bug_chart.py create mode 100644 modules/last_comments.py create mode 100644 modules/login.py create mode 100644 modules/menu.py diff --git a/hms.py b/hms.py new file mode 100644 index 0000000..cc4569a --- /dev/null +++ b/hms.py @@ -0,0 +1,6 @@ + +import modules +from hooks import * +from themes import * +from generate import * +modules.init() diff --git a/modules.py b/modules.py index 422bc9d..a8ddac6 100644 --- a/modules.py +++ b/modules.py @@ -6,6 +6,7 @@ def init(): ... Here goes definitions. """ +import sys import os import os.path @@ -14,7 +15,7 @@ from themes import * from generate import * MODS_DIR = "modules" -modules = {} +modules = [] exports = ['export', 'export_as', 'hook', 'intercept','themeable','theme', 'generates', 'generate', 'Generated'] @@ -32,21 +33,28 @@ def export_as(obj,name): exports.append(name) @hook -def module_load(name): - global modules +def module_load(mod): global exports - mod = modules[name] for n in exports: setattr(mod,n,globals()[n]) - modules[name].init() + mod.init() def init(): - for name in modules: - module_load(name) + for w,mod in modules: + module_load(mod) -for dirpath, dirs, files in os.walk(MODS_DIR): +for dirpath, dirs, files in os.walk(os.path.join(os.path.dirname(__file__),MODS_DIR)): for file in files: if file[-3:] == ".py": name = os.path.join(dirpath,file[:-3]) - module = __import__(name) - modules[name] = module + try: + sys.path.append(os.path.dirname(name)) + module = __import__(os.path.basename(name),{},{},['']) + try: + weight = module.weight + except AttributeError: + weight = 0 + except ImportError,e: + raise ImportError,"Could not import module '%s': %s" % (name,e) + modules.append((weight,module)) +modules.sort() diff --git a/modules/blocks.py b/modules/blocks.py index 95732a6..ea785c2 100644 --- a/modules/blocks.py +++ b/modules/blocks.py @@ -1,4 +1,6 @@ +#encoding: utf-8 +weight = -10 def init(): print "Blocks module loaded." @@ -8,12 +10,12 @@ def init(): @themeable def block(blk): - return """
+ return u"""

%s

%s
-
""" % (blk.title,blk.content) +
""" % (blk.name,blk.title,blk.content) class Block(Generated): name = "Block" diff --git a/modules/bug_chart.py b/modules/bug_chart.py new file mode 100755 index 0000000..1867947 --- /dev/null +++ b/modules/bug_chart.py @@ -0,0 +1,74 @@ +#!/usr/bin/python +#encoding: utf-8 + +if __name__ == '__main__': + export = lambda f: f + +def init(): + import os.path + import math + import cairo + + root = "/home/portnov/www/projects/" + labels = [u"Неподтвержденые",u"Подтвержденные",u"В работе",u"Исправлены",u"Неактуальны"] + colors = [(0.75,0.75,1), + (1,0.75,0.75), + (1,1,0.75), + (0.75,1,0.75), + (0.75,0.75,0.75)] +# labels = ["U","P","W","F","N/A"] + width,height = 200,300 + + def piechart(data,outfile): + surf = cairo.ImageSurface(cairo.FORMAT_ARGB32,width,height) + cx = cairo.Context(surf) + centerx,centery = width/2, width/2 + radius = 0.9*min(width,height)/2 + s = sum(data) + n = len(data) + current_angle = 0 + cx.set_source_rgb(255,255,255) + cx.rectangle(0,0,width,height) + cx.fill() + i=0 + cx.select_font_face("DejaVu Sans", cairo.FONT_SLANT_NORMAL, cairo.FONT_WEIGHT_NORMAL) + cx.set_font_size(12) + + for d in data: + angle = float(d)*math.pi*2.0/float(s) + end_angle = current_angle + angle + cx.new_path() + cx.set_source_rgb(*colors[i]) + cx.move_to(centerx,centery) + cx.arc(centerx,centery,radius,current_angle,end_angle) + cx.line_to(centerx,centery) + cx.close_path() + current_angle = end_angle + cx.fill() + cx.set_source_rgb(*colors[i]) + y = height-20*n+20*i + cx.rectangle(15,y,15,15) + cx.fill() + cx.set_source_rgb(0,0,0) + cx.move_to(40,y+12) + cx.show_text(labels[i]) + cx.stroke() + i += 1 + surf.write_to_png(outfile) + + def chart_path(data): + return "/media/charts/chart_" + "_".join(map(str,data)) + ".png" + + @export + def bug_chart(data): + path = chart_path(data) + if not os.path.exists(root+path): + piechart(data,root+path) + return "Bugs chart" % path + + if __name__ == '__main__': + piechart([0,0,1,0,0],'chart.png') + +if __name__ == '__main__': + init() + diff --git a/modules/last_comments.py b/modules/last_comments.py new file mode 100644 index 0000000..702286c --- /dev/null +++ b/modules/last_comments.py @@ -0,0 +1,27 @@ +#encoding: utf-8 + +def init(): + + from mgmt.models import Comment + + @block + def last_comments(): + def short_comment(comment): + def get_url(comment): + type = comment.object_type + id = comment.object_id + if type=='Project': + return "/projects/%s/#comment-%s" % (id,comment.id) + if type=='Bug': + return "/bugs/%s/#comment-%s" % (id,comment.id) + if type=='Document': + return "/docs/%d/#comment-%s" % (id,comment.id) + return "
  • %s
    • " % (get_url(comment),comment.title) + b = Block() + b.name='last_comments' + b.title = u"Последние комментарии" + comments = Comment.objects.order_by('-created')[:10] + b.content = "\n" + return b + + diff --git a/modules/login.py b/modules/login.py new file mode 100644 index 0000000..66d761f --- /dev/null +++ b/modules/login.py @@ -0,0 +1,17 @@ +#encoding: utf-8 + +def init(): + + @block + def login_block(): + b = Block() + b.name = 'login' + b.title = u'Вход в систему' + b.content = u""" +
    +

    +

    + + +
    """ + return b diff --git a/modules/menu.py b/modules/menu.py new file mode 100644 index 0000000..86b9a25 --- /dev/null +++ b/modules/menu.py @@ -0,0 +1,14 @@ +#encoding: utf-8 + +def init(): + + @export + def menu(request): + m = [] + m.append((u'Главная','/')) + if request.user.is_authenticated(): + m.append((u'Выход','/logout/')) + else: + m.append((u'Вход','/login/')) + L = ["%s" % (l,t) for t,l in m] + return L -- 1.7.2.3