-
Notifications
You must be signed in to change notification settings - Fork 0
/
models.py
41 lines (35 loc) · 993 Bytes
/
models.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import json
from peewee import *
db = SqliteDatabase('violence.db')
class BaseModel(Model):
class Meta:
database = db
class Article(BaseModel):
"""
An article is a single incident as crawled from the reach-out webpage
"""
date = DateField(index=True)
month_only = BooleanField(default=False)
place = CharField()
description = TextField()
hash = BlobField(index=True)
class Location(BaseModel):
"""
A location describes the place an incident has happened
"""
confidence = IntegerField()
lat = DoubleField()
lng = DoubleField()
match = CharField()
returned_place = CharField()
article = ForeignKeyField(Article)
class Category(BaseModel):
"""
Describes the category of an incident (e.g. sexism, racism, antisemitism etc)
"""
name = CharField()
article = ForeignKeyField(Article)
# Set up the tables
def create_tables():
db.connect()
db.create_tables([Article, Location, Category])