class LinkPlanet(Planet):
"""A URL attachment"""
id = db.Column(db.String(32), ForeignKey('planet.id'), primary_key=True)
url = db.Column(db.Text)
__mapper_args__ = {'polymorphic_identity': 'link'}
class PicturePlanet(Planet):
"""A Picture attachment"""
id = db.Column(db.String(32), ForeignKey('planet.id'), primary_key=True)
filename = db.Column(db.Text)
__mapper_args__ = {'polymorphic_identity': 'picture'}
class Word(Serializable, db.Model):
__tablename__ = "words"
word = db.Column(db.String(32), , primary_key=True)
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
collection_id = db.Column(db.Integer, db.ForeignKey('word_collections.id'))
def create_word():
pass
class Notification(db.Model):
"""Represents a stored notification to the user"""
id = db.Column(db.String(32), primary_key=True)
kind = db.Column(db.String(32))
created = db.Column(db.DateTime, default=datetime.datetime.now())
to_persona_id = db.Column(db.String(32))
def __init__(self, kind, to_persona_id):
from uuid import uuid4
self.id = uuid4().hex
self.kind = kind
self.to_persona_id = to_persona_id
class Star(Serializable, db.Model):
"""A Star represents a post"""
__tablename__ = "star"
id = db.Column(db.String(32), primary_key=True)
text = db.Column(db.Text)
created = db.Column(db.DateTime, default=datetime.datetime.now())
modified = db.Column(db.DateTime,
default=datetime.datetime.now(),
onupdate=datetime.datetime.now())
planets = db.relationship(
'Planet',
secondary='satellites',
backref=db.backref('starmap'),
primaryjoin="satellites.c.star_id==star.c.id",
secondaryjoin="satellites.c.planet_id==planet.c.id")
creator = db.relationship('Persona',
backref=db.backref('starmap'),
primaryjoin="Persona.id==Star.creator_id")
creator_id = db.Column(db.String(32), db.ForeignKey('persona.id'))
def __init__(self, id, text, creator):
self.id = id
# TODO: Attach multiple items as 'planets'
self.text = text
if not isinstance(creator, Persona):
self.creator_id = creator
else:
self.creator_id = creator.id
def __repr__(self):
ascii_text = self.text.encode('utf-8')
return "<Star {}: {}>".format(
self.creator_id[:6],
(ascii_text[:24] if len(ascii_text) <= 24 else ascii_text[:22] +
".."))
def get_absolute_url(self):
return url_for('star', id=self.id)
def hot(self):
"""i reddit"""
from math import log
# Uncomment to assign a score with analytics.score
#s = score(self)
s = 1.0
order = log(max(abs(s), 1), 10)
sign = 1 if s > 0 else -1 if s < 0 else 0
return round(order + sign * epoch_seconds(self.created) / 45000, 7)
class Tag(Serializable, db.Model):
__tablename__ = "tags"
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
word_collection_id = db.Column(db.Integer, db.ForeignKey('word_collections.id'), primary_key=True, nullable=False, autoincrement=True)
count = db.Column(db.Integer)
def create_tag():
pass
def update_count(count):
pass
def get_count():
pass
class Orb(db.Model):
"""Stub for any object that might exist in a starmap"""
__tablename__ = 'orb'
id = db.Column(db.String(32), primary_key=True)
type = db.Column(db.String(32))
modified = db.Column(db.DateTime)
creator = db.Column(db.String(32))
def __init__(self, object_type, id, modified, creator=None):
self.id = id
self.type = object_type
self.modified = modified
self.creator = creator
def __repr__(self):
return "<Orb:{} {}>".format(self.type, self.id[:6])
class Word_Collection(Serializable, db.Model):
__tablename__ = "word_collections"
id = db.Column(db.Integer, primary_key=True, nullable=False, autoincrement=True)
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
def create_collection():
pass
def update_counts(tags, counts):
pass
def get_counts(tags):
pass
def add_word(word):
pass
def get_words():
pass
class Planet(Serializable, db.Model):
"""A Planet represents an attachment"""
__tablename__ = 'planet'
id = db.Column(db.String(32), primary_key=True)
title = db.Column(db.Text)
kind = db.Column(db.String(32))
created = db.Column(db.DateTime, default=datetime.datetime.now())
modified = db.Column(db.DateTime,
default=datetime.datetime.now(),
onupdate=datetime.datetime.now())
source = db.Column(db.String(128))
__mapper_args__ = {
'polymorphic_identity': 'planet',
'polymorphic_on': kind
}
class Starmap(db.Model):
__tablename__ = 'starmap'
id = db.Column(db.String(32), primary_key=True)
index = db.relationship(
'Orb',
secondary='starmap_index',
primaryjoin='starmap_index.c.starmap_id==starmap.c.id',
secondaryjoin='starmap_index.c.orb_id==orb.c.id')
def __init__(self, id):
self.id = id
def __contains__(self, key):
return (key in self.index)
def __repr__(self):
return "<Starmap {}>".format(self.id)
def add(self, orb):
"""Add Orb to this starmap"""
if orb in self.index:
raise KeyError("{} is already part of {}.".format(orb, self))
return self.index.append(orb)
# data field needs to be decoded if the message is signed
message.data = json.loads(message.data)
return message
def sign(self, author):
"""Sign a message using an author persona. Make sure not to change message data after signing"""
if not isinstance(self.data, str):
self.data = json.dumps(self.data)
self.signature = author.sign(self.data)
self.author_id = author.id
self.send_attributes.extend(["signature", "author_id"])
t_starmap_index = db.Table(
'starmap_index',
db.Column('starmap_id', db.String(32), db.ForeignKey('starmap.id')),
db.Column('orb_id', db.String(32), db.ForeignKey('orb.id'))
)
class Starmap(db.Model):
__tablename__ = 'starmap'
id = db.Column(db.String(32), primary_key=True)
index = db.relationship(
'Orb',
secondary='starmap_index',
primaryjoin='starmap_index.c.starmap_id==starmap.c.id',
secondaryjoin='starmap_index.c.orb_id==orb.c.id')
def __init__(self, id):
self.id = id
class Persona(Serializable, db.Model):
"""A Persona represents a user profile"""
__tablename__ = "persona"
id = db.Column(db.String(32), primary_key=True)
username = db.Column(db.String(80))
email = db.Column(db.String(120))
contacts = db.relationship(
'Persona',
secondary='contacts',
primaryjoin='contacts.c.left_id==persona.c.id',
secondaryjoin='contacts.c.right_id==persona.c.id')
crypt_private = db.Column(db.Text)
crypt_public = db.Column(db.Text)
sign_private = db.Column(db.Text)
sign_public = db.Column(db.Text)
modified = db.Column(db.DateTime,
default=datetime.datetime.now(),
onupdate=datetime.datetime.now())
soma_id = db.Column(db.String(32), db.ForeignKey('starmap.id'))
soma = db.relationship('Starmap',
backref="personas",
primaryjoin='starmap.c.id==persona.c.soma_id')
def __init__(self,
id,
username,
email=None,
sign_private=None,
sign_public=None,
crypt_private=None,
crypt_public=None):
self.id = id
self.username = username
self.email = email
self.sign_private = sign_private
self.sign_public = sign_public
self.crypt_private = crypt_private
self.crypt_public = crypt_public
def __repr__(self):
return "<{} [{}]>".format(str(self.username), self.id[:6])
def get_email_hash(self):
"""Return sha256 hash of this user's email address"""
return sha256(self.email).hexdigest()
def get_absolute_url(self):
return url_for('persona', id=self.id)
def generate_keys(self, password):
""" Generate new RSA keypairs for signing and encrypting. Commit to DB afterwards! """
# TODO: Store keys encrypted
rsa1 = RsaPrivateKey.Generate()
self.sign_private = str(rsa1)
self.sign_public = str(rsa1.public_key)
rsa2 = RsaPrivateKey.Generate()
self.crypt_private = str(rsa2)
self.crypt_public = str(rsa2.public_key)
def encrypt(self, data):
""" Encrypt data using RSA """
key_public = RsaPublicKey.Read(self.crypt_public)
return key_public.Encrypt(data)
def decrypt(self, cypher):
""" Decrypt cyphertext using RSA """
key_private = RsaPrivateKey.Read(self.crypt_private)
return key_private.Decrypt(cypher)
def sign(self, data):
""" Sign data using RSA """
from base64 import b64encode
key_private = RsaPrivateKey.Read(self.sign_private)
signature = key_private.Sign(data)
return b64encode(signature)
def verify(self, data, signature_b64):
""" Verify a signature using RSA """
from base64 import b64decode
signature = b64decode(signature_b64)
key_public = RsaPublicKey.Read(self.sign_public)
return key_public.Verify(data, signature)
import os
from flask import url_for
from hashlib import sha256
from keyczar.keys import RsaPrivateKey, RsaPublicKey
from web_ui import db
from web_ui.helpers import Serializable, epoch_seconds
from sqlalchemy import ForeignKey
from sqlalchemy.exc import OperationalError
#
# Setup follower relationship on Persona objects
#
t_contacts = db.Table(
'contacts', db.Column('left_id', db.String(32),
db.ForeignKey('persona.id')),
db.Column('right_id', db.String(32), db.ForeignKey('persona.id')))
class Persona(Serializable, db.Model):
"""A Persona represents a user profile"""
__tablename__ = "persona"
id = db.Column(db.String(32), primary_key=True)
username = db.Column(db.String(80))
email = db.Column(db.String(120))
contacts = db.relationship(
'Persona',
secondary='contacts',
primaryjoin='contacts.c.left_id==persona.c.id',
secondaryjoin='contacts.c.right_id==persona.c.id')
class Naive_Bayes_Tag_Model(Serializable, db.Model):
__tablename__ = "naive_bayes_model"
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
tag_id = db.Column(db.Integer, db.ForeignKey(tags.id))
model = db.Column(db.PickleType)
def update_counts(tags, counts):
pass
def get_counts(tags):
pass
def add_word(word):
pass
def get_words():
pass
class Word_under_Tag_Count(Serializable, db.Model):
__tablename__ = "word_under_tag_counts"
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
collection_id = db.Column(db.Integer, db.ForeignKey('word_collections.id'), primary_key=True, nullable=False, autoincrement=False)
tag_id = db.Column(db.Integer, db.ForeignKey('tags.id'), primary_key=True, nullable=False, autoincrement=False)
count = db.Column(db.Integer)
class Tag(Serializable, db.Model):
__tablename__ = "tags"
persona = db.Column(db.String(32), db.ForeignKey('persona.id'), primary_key=True)
word_collection_id = db.Column(db.Integer, db.ForeignKey('word_collections.id'), primary_key=True, nullable=False, autoincrement=True)
count = db.Column(db.Integer)
def create_tag():
pass
def update_count(count):
pass
