loc-sas/testdata.py at master · GovWizely/loc-sas · 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
import logging
import random
import uuid
from datetime import datetime

from app import appbuilder, db
from app.models import Author, Claimant, CopyrightApplication

log = logging.getLogger(__name__)


def get_random_name(names_list, size=1):
    name_lst = [
        names_list[random.randrange(0, len(names_list))].decode("utf-8").capitalize()
        for _ in range(0, size)
    ]
    return " ".join(name_lst)


role_user = appbuilder.sm.find_role("User")
users_exist = appbuilder.sm.find_user("user1")

if not users_exist:
    user1 = appbuilder.sm.add_user(
        "user1", "user1", "test", "user1@example.org", role_user, "password"
    )
    user2 = appbuilder.sm.add_user(
        "user2", "user2", "test", "user2@example.org", role_user, "password"
    )
    user3 = appbuilder.sm.add_user(
        "user3", "user3", "test", "user3@example.org", role_user, "password"
    )

    f = open("NAMES.DIC", "rb")
    names_list = [x.strip() for x in f.readlines()]

    f.close()

    j = 1
    for i in range(1, 75):
        c = CopyrightApplication()
        c.primary_title = get_random_name(names_list, random.randrange(2, 5))
        c.year_completed = random.choice(range(2013, 2019))
        name = c.primary_title.split()
        a = Author()
        a.copyright_application_id = i
        a.first_name = name[0]
        a.last_name = name[-1]
        a.citizenship = random.choice(["US", "GB", "CA", "MX", "JP"])
        a.domicile = random.choice(["GR", "IT", "KY", "MD"])
        a.year_of_birth = "1990"
        a.organization = False
        c.authors = [a]
        cl = Claimant()
        cl.copyright_application_id = i
        cl.organization = False
        cl.first_name = name[0]
        cl.last_name = name[-1]
        cl.address = "3333 Fairfax Dr"
        cl.city = "Arlington"
        cl.state = "VA"
        cl.postal_code = "22203"
        cl.country = "US"
        c.claimants = [cl]
        month = random.choice(range(1, 12))
        day = random.choice(range(1, 28))
        c.created_on = datetime(int(c.year_completed), month, day)
        c.changed_on = datetime.now()
        if j == 1:
            j += 1
            _user = user1
        elif j == 2:
            j += 1
            _user = user2
        else:
            j = 1
            _user = user3
        c.created_by = _user
        c.changed_by = _user
        c.service_request_id = str(uuid.uuid1())
        c.application_status = random.choice(["draft", "under_review", "approved"])
        c.work_deposit_name = "The Best Book.pdf"
        c.work_deposit_url = "https://some-cloud-storage.io/us-copyright-office/the_best_book.pdf"
        db.session.add(c)
        try:
            db.session.commit()
            print("inserted", c)
        except Exception as e:
            log.error("CopyrightApplication creation error: %s", e)
            db.session.rollback()