OpenData500/models.py at master · GovLab/OpenData500 · GitHub

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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#!/usr/bin/env python
# -*- coding: utf8 -*-
import os, sys
from mongoengine import *
from datetime import datetime

class Person(EmbeddedDocument):
	firstName = StringField()
	lastName = StringField()
	title = StringField()
	email = StringField()
	phone = StringField()
	org = StringField() # <--------DEPRECATE
	contacted = BooleanField() # <--------DEPRECATE

class Company(Document):
	companyName = StringField()
	prettyName = StringField()
	url = StringField()
	contact = EmbeddedDocumentField(Person)
	ceo = EmbeddedDocumentField(Person)
	yearFounded = StringField()
	city = StringField()
	state = StringField()
	country = StringField()
	zipCode = StringField()
	fte = StringField()
	companyType = StringField() #Public, Private, etc
	companyCategory = StringField() #Categories
	revenueSource = ListField(StringField()) #checkbox with options
	businessModel = ListField(StringField())
	socialImpact = ListField(StringField())
	description = StringField()
	descriptionShort = StringField()
	financialInfo = StringField()
	sourceCount = StringField() #From how many sources does your company use data?
	dataTypes = ListField(StringField())
	dataComments = StringField() #Please give comments, good or bad....
	exampleUses = StringField()
	dataImpacts = ListField(StringField())
	dataWishlist = StringField()
	agencies = ListField(ReferenceField('Agency'))
	ts = ComplexDateTimeField(default=datetime.now())
	lastUpdated = ComplexDateTimeField()
	display = BooleanField() #Display on site
	submittedSurvey = BooleanField()
	vetted = BooleanField() #vetted by us
	vettedByCompany = BooleanField() #vetted by them
	submittedThroughWebsite = BooleanField() #submitted through website
	locked = BooleanField() #Locked from public editing
	notes = StringField() #notes by admin
	filters = ListField(StringField()) #Filters for the full list page.

	@queryset_manager
	def objects(doc_cls, queryset):
		return queryset.order_by('prettyName')

class Agency(Document):
	ts = ComplexDateTimeField(default=datetime.now())
	name = StringField()
	abbrev = StringField()
	prettyName = StringField()
	url = StringField() #for whatever is more specific, agency or subagency
	source = StringField() #DATA.GOV WEBSITE OR USER INPUT
	subagencies = ListField(EmbeddedDocumentField('Subagency'))
	dataType = StringField() #Federal, State, City/County, Other
	datasets = ListField(EmbeddedDocumentField('Dataset'))
	country = StringField()
	usedBy = ListField(ReferenceField(Company))
	usedBy_count = IntField()
	notes = StringField()

class Subagency(EmbeddedDocument):
	name = StringField()
	abbrev = StringField()
	url = StringField()
	datasets = ListField(EmbeddedDocumentField('Dataset'))
	usedBy = ListField(ReferenceField(Company))

class Dataset(EmbeddedDocument):
	datasetName = StringField()
	datasetURL = StringField()
	rating = IntField()
	usedBy = ReferenceField(Company)

class Users(Document):
	ts = ComplexDateTimeField(default=datetime.now())
	username = StringField()
	password = StringField()
	country = StringField()
	permissions = StringField()

class States(EmbeddedDocument):
	name = StringField()
	abbrev = StringField()
	count = IntField()

class Stats(Document):
	lastUpdate = ComplexDateTimeField()
	totalCompanies = IntField() #total
	totalCompaniesWeb = IntField() #submitted through web survey
	totalCompaniesSurvey = IntField() #completed surveys
	totalCompaniesDisplayed = IntField() #displayed companies
	totalAgencies = IntField()
	states = ListField(EmbeddedDocumentField(States))
	country = StringField()

class Visit(Document):
	ts = ComplexDateTimeField(default=datetime.now())
	r = StringField() #referer
	p = StringField() #page
	ua = StringField() #userAgent
	ip = StringField()

Agency.register_delete_rule(Company, "agencies", PULL)
Company.register_delete_rule(Agency, 'usedBy', PULL)