include pylons_app/config/deployment.ini_tmpl

include hg_app_daemon

include hg_app_daemon2

recursive-include pylons_app/public/css *

recursive-include pylons_app/public/images *

include pylons_app/public/js/yui2.js

include pylons_app/public/js/excanvas.min.js

include pylons_app/public/js/yui.flot.js

recursive-include pylons_app/templates *

from sqlalchemy.interfaces import ConnectionProxy

import time

import logging

log = logging.getLogger('timerproxy')

BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = xrange(30, 38)

def color_sql(sql):

    COLOR_SEQ = "\033[1;%dm"

    COLOR_SQL = YELLOW

    normal = '\x1b[0m'

    return COLOR_SEQ % COLOR_SQL + sql + normal 

def one_space_trim(s):

    if s.find("  ") == -1:

        return s

    else:

        s = s.replace('  ', ' ')

        return one_space_trim(s)

def format_sql(sql):

    sql = color_sql(sql)

    sql = sql.replace('\n', '')

    sql = one_space_trim(sql)

    sql = sql\

        .replace('SELECT', '\n\tSELECT \n\t')\

        .replace('UPDATE', '\n\tUPDATE \n\t')\

        .replace('DELETE', '\n\tDELETE \n\t')\

        .replace('FROM', '\n\tFROM')\

        .replace('ORDER BY', '\n\tORDER BY')\

        .replace('LIMIT', '\n\tLIMIT')\

        .replace('WHERE', '\n\tWHERE')\

        .replace('AND', '\n\tAND')\

        .replace('LEFT', '\n\tLEFT')\

        .replace('INNER', '\n\tINNER')\

        .replace('INSERT', '\n\tINSERT')\

        .replace('DELETE', '\n\tDELETE')

    return sql

class TimerProxy(ConnectionProxy):

    def cursor_execute(self, execute, cursor, statement, parameters, context, executemany):

        now = time.time()

        try:

            log.info(">>>>> STARTING QUERY >>>>>")

            return execute(cursor, statement, parameters, context)

        finally:

            total = time.time() - now

            try:

    name = Column("name", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

    lastname = Column("lastname", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

    email = Column("email", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

    last_login = Column("last_login", DATETIME(timezone=False), nullable=True, unique=None, default=None)

    user_log = relation('UserLog')

    user_perms = relation('UserToPerm', primaryjoin="User.user_id==UserToPerm.user_id")

    @LazyProperty

    def full_contact(self):

        return '%s %s <%s>' % (self.name, self.lastname, self.email)

    def __repr__(self):

        return "<User('id:%s:%s')>" % (self.user_id, self.username)

    def update_lastlogin(self):

        """Update user lastlogin"""

        import datetime

        try:

            session = Session.object_session(self)

            self.last_login = datetime.datetime.now()

            session.add(self)

            session.commit()

        except Exception:

            session.rollback()        

class UserLog(Base): 

    __tablename__ = 'user_logs'

    __table_args__ = {'useexisting':True}

    user_log_id = Column("user_log_id", INTEGER(), nullable=False, unique=True, default=None, primary_key=True)

    user_id = Column("user_id", INTEGER(), ForeignKey(u'users.user_id'), nullable=False, unique=None, default=None)

    user_ip = Column("user_ip", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None) 

    repository = Column("repository", TEXT(length=None, convert_unicode=False, assert_unicode=None), ForeignKey(u'repositories.repo_name'), nullable=False, unique=None, default=None)

    action = Column("action", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

    action_date = Column("action_date", DATETIME(timezone=False), nullable=True, unique=None, default=None)

    user = relation('User')

class Repository(Base):

    __tablename__ = 'repositories'

    __table_args__ = (UniqueConstraint('repo_name'), {'useexisting':True},)

    repo_id = Column("repo_id", INTEGER(), nullable=False, unique=True, default=None, primary_key=True)

    repo_name = Column("repo_name", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=False, unique=True, default=None)

    user_id = Column("user_id", INTEGER(), ForeignKey(u'users.user_id'), nullable=False, unique=False, default=None)

    private = Column("private", BOOLEAN(), nullable=True, unique=None, default=None)

    description = Column("description", TEXT(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

from pylons_app.tests import *

class TestLoginController(TestController):

    def test_index(self):

        response = self.app.get(url(controller='login', action='index'))

        assert response.status == '200 OK','Wrong response from login page'

        # Test response...

    def test_login_admin_ok(self):

        response = self.app.post(url(controller='login', action='index'),

                                 {'username':'test_admin',

                                  'password':'test'})

        assert response.status == '302 Found','Wrong response code from login'

    def test_login_regular_ok(self):

        response = self.app.post(url(controller='login', action='index'),

                                  'password':'test'})

        assert response.status == '302 Found','Wrong response code from login'

    def test_login_ok_came_from(self):

        test_came_from = '/_admin/users'

        response = self.app.post(url(controller='login', action='index',came_from=test_came_from),

                                 {'username':'test_admin',

                                  'password':'test'})

        assert response.status == '302 Found','Wrong response code from came from redirection'

        response = response.follow()

        assert response.status == '200 OK','Wrong response from login page'

        assert 'Users administration' in response.body,'No proper title in response'

    def test_login_wrong(self):

        response = self.app.post(url(controller='login', action='index'),

                                 {'username':'error',

                                  'password':'test'})

        assert response.status == '200 OK','Wrong response from login page'

        assert 'invalid user name' in response.body,'No error username message in response'

        assert 'invalid password' in response.body,'No error password message in response'