junmakii_simple_blogging.py

#!/usr/bin/python
#-*- coding: utf-8 -*-
import os, sys, datetime
import argparse
import json
import jinja2
from sqlalchemy.types import (VARCHAR, TEXT, BOOLEAN, INTEGER, FLOAT, TIME,
                              DATE, DATETIME, TIMESTAMP)
from sqlalchemy.schema import (ForeignKey)
from sqlalchemy.sql.expression import (asc, desc)
from sqlalchemy.schema import Column
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, scoped_session

Base = declarative_base()

file_path = os.path.abspath(__file__)
base_name = os.path.dirname(file_path)
template_name = 'junmakii_simple_blogging.html'
single_page_template_name = 'junmakii_simple_blogging.single.html'
template_dirs = [os.path.join(os.path.dirname(file_path), 'templates')]
template_env = jinja2.Environment(loader=jinja2.FileSystemLoader(template_dirs),
                                  autoescape=True,
                                  extensions=['jinja2.ext.autoescape'])
template = template_env.get_template(template_name)


class Text(Base):
    __tablename__ = 'texts'
    id = Column(INTEGER, primary_key=True, autoincrement=True)
    name = Column(VARCHAR(255), nullable=True)
    text = Column(TEXT, nullable=False)
    keyword = Column(VARCHAR(255), nullable=True)
    created_at = Column(DATETIME, default=datetime.datetime.now())


db_path = os.path.join(os.environ.get('XDG_HOME') or os.environ.get('HOME'),
                       '.' + os.path.basename(__file__) + '.db.sqlite')
db_url = 'sqlite:///' + db_path

print(Text.__table__.columns.keys())


def parse_args(argv):
    parser = argparse.ArgumentParser(description='')
    parser.add_argument('-d', '--database', default=db_url, action='store')
    parser.add_argument('--initialize', action='store_true')
    parser.add_argument('-a', '--all', action='store_true')
    parser.add_argument('-l', '--limit', type=int, default=10, action='store')
    parser.add_argument('-s', '--search', default='%', action='store')
    parser.add_argument('-f', '--file', action='store')
    parser.add_argument('-r', '--remove', type=int, action='store')
    parser.add_argument('-e', '--export', action='store')
    parser.add_argument('-n', '--name', default='', action='store')
    parser.add_argument('-k', '--keyword', default='', action='store')
    parser.add_argument('-t', '--text', action='store')
    parser.add_argument('--html', action='store')
    args = parser.parse_args(argv)
    return args


def datetime_json_format(obj):
    return obj.isoformat() if hasattr(obj, 'isoformat') else obj


def get_db(db_url):
    engine = create_engine(db_url)
    return engine


def write_html_single_page(output, a):
    template = template_env.get_template(single_page_template_name)
    for i in range(len(a)):
        dest = os.path.join(output, str(a[i].id) + '.html')
        data = template.render(page=a[i])
        with open(dest, 'w') as fp:
            fp.write(data.encode('utf8'))
        print(dest)


def write_html(db, output, per=5):
    session = scoped_session(sessionmaker(bind=db))
    texts = session.query(Text).order_by(desc(Text.created_at)).all()
    write_html_single_page(output, texts)
    length = len(texts) / per
    for i in range(length + 1):
        j = i + 1
        selected_texts = texts[i * per:i * per + per]
        dest = os.path.join(output, str(j) + '-index.html')
        data = template.render({"items": selected_texts, "current_page": j})
        with open(dest, 'w') as fp:
            fp.write(data.encode('utf8'))
        print(dest)


def main(argv):
    args = parse_args(argv)
    db = get_db(args.database)
    session = scoped_session(sessionmaker(bind=db))
    if args.html:
        write_html(db, args.html)
        return 0
    if args.initialize:
        Base.metadata.create_all(db)
        return 0
    if args.remove:
        text = session.query(Text).filter(Text.id == args.remove)[0]
        session.delete(text)
        session.commit()
        return 0
    if args.all:
        texts = session.query(Text).filter(Text.text.like(
            args.search)).order_by(desc(Text.created_at))[0:args.limit]
        for text in texts:
            for key in Text.__table__.columns.keys():
                print('key: ' + key)
                print(getattr(text, key))
            print('---')
        text_dicts = dict([(key, getattr(text, key))
                           for key in Text.__table__.columns.keys()])
        if args.export:
            json_str = json.dumps(text_dicts, default=datetime_json_format)
            with open(args.export, 'w') as fp:
                fp.write(json_str.encode('utf8'))
        return 0
    if args.file:
        text = open(args.file, 'r').read()
        name = args.file
    else:
        text = args.text
        name = args.name
    keyword = args.keyword
    text = Text(text=text.decode('utf8'),
                name=name.decode('utf8'),
                keyword=keyword.decode('utf8'))
    print(session.add(text))
    print(session.commit())


if __name__ == '__main__':
    main(sys.argv[1:])