• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• dbpy-master
  • samples
    • app.py
    • model.py
    • orm.py
    • schema.sql
  • db
    • dialect.py
    • pool.py
    • _db.py
    • pymysql
      • dialect.py
      • connection.py
      • __init__.py
    • mysql
      • dialect.py
      • connection.py
      • __init__.py
    • query
      • select.py
      • update.py
      • insert.py
      • __init__.py
      • expr.py
      • delete.py
      • base.py
    • connection.py
    • __init__.py
    • errors.py
  • CHANGES
  • LICENSE
  • README_CN.rst
  • .gitattributes
  • README.rst
  • setup.py
  • MAUNIFEST.in
  • setup.cfg
  • tests
    • dbt.py
    • test.sql
    • pymysqlt.py
  • .gitignore

#!/usr/bin/env python
# Copyright (C) 2014-2015 Thomas Huang
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from hashlib import sha224
from datetime import datetime


class User(object):

    def __init__(self, username, email, real_name, password, bio, status, role='user', uid=None):
        """If the user load from database, if will intialize the uid and secure password.
        Otherwise will hash encrypt the real password

        arg role enum: the string in ('user', 'editor', 'administrator')
        arg status enum: the string in ('actived', 'inactive')
        arg password fix legnth string: the use sha224 password hash
        """

        self.username = username
        self.email = email
        self.real_name = real_name
        self.bio = bio
        self.status = status
        self.role = role

        if uid:
            self.uid = uid
            self.password = password
        else:
            self.password = self.secure_password(password)

    def check(self, password):
        """Check the password"""
        return self.password == self.secure_password(password)

    def secure_password(self, password):
        """Encrypt password to sha224 hash"""
        return sha224(password).hexdigest()

    def as_json(self):
        data = self.__dict__.copy()
        del data['password']
        return data


class Post(object):

    def __init__(self, title, slug, description, html, css, js, category, status,
                 comments, author=None, created=datetime.now(), pid=None):
        self.title = title
        self.slug = slug
        self.description = description
        self.created = created
        self.html = html
        self.css = css
        self.js = js
        self.category = category
        self.status = status
        self.comments = comments
        self.author = author
        self.created = created
        self.pid = pid

    def as_json(self):
        data = self.__dict__.copy()
        return data