1
=== modified file 'desktopcouch/contacts/testing/create.py'
2
--- desktopcouch/contacts/testing/create.py	2009-08-19 10:07:54 +0000
3
+++ desktopcouch/contacts/testing/create.py	2009-09-04 08:59:18 +0000
4
@@ -30,25 +30,55 @@
5
30
    'Daniel', 'William', 'James', 'Alfie', 'Grace', 'Ruby', 'Olivia',
30
    'Daniel', 'William', 'James', 'Alfie', 'Grace', 'Ruby', 'Olivia',
6
31
    'Emily', 'Jessica', 'Sophie', 'Chloe', 'Lily', 'Ella', 'Amelia')
31
    'Emily', 'Jessica', 'Sophie', 'Chloe', 'Lily', 'Ella', 'Amelia')
7
32
LAST_NAMES = ('Dalglish', 'Grobbelaar', 'Lawrenson', 'Beglin', 'Nicol',
32
LAST_NAMES = ('Dalglish', 'Grobbelaar', 'Lawrenson', 'Beglin', 'Nicol',
9
33
    'Whelan', 'Hansen', 'Johnston', 'Rush', 'Molby', 'MacDonald', 'McMahon')
33
    'Whelan', 'Hansen', 'Johnston', 'Rush', 'Molby', 'MacDonald', 'McMahon',
10
34
    'Penny', 'Leicester', 'Langley', 'Commodore', 'Touchstone', 'Fielding')
11
35
COMPANIES = ('Gostram', 'Grulthing', 'Nasform', 'Smarfish', 'Builsank')
12
36
STREET_TYPES = ('Street', 'Road', 'Lane', 'Avenue', 'Square', 'Park', 'Mall')
13
34
CITIES = ('Scunthorpe', 'Birmingham', 'Cambridge', 'Durham', 'Bedford')
37
CITIES = ('Scunthorpe', 'Birmingham', 'Cambridge', 'Durham', 'Bedford')
14
35
COUNTRIES = ('England', 'Ireland', 'Scotland', 'Wales')
38
COUNTRIES = ('England', 'Ireland', 'Scotland', 'Wales')
15
36
39
19
37
def random_string(length=10):
40
def head_or_tails():
20
38
    """Return a string of random lowercase letters and of specified length"""
41
    """Randomly return True or False"""
21
39
    return ''.join(random.sample(string.lowercase, length))
42
    return random.choice((False, True))
22
43
23
44
def random_bools(num_bools, at_least_one_true=True):
24
45
    """
25
46
    Return a sequence of random booleans. If at_least_one_true,
26
47
    guarantee at list one True value.
27
48
    """
28
49
    if num_bools < 2:
29
50
        raise RuntimeError('Cannot build a sequence smaller than two elements')
30
51
    bools = [head_or_tails() for __ in range(num_bools)]
31
52
    if at_least_one_true and not any(bools):
32
53
        bools[random.randrange(num_bools)] = True
33
54
    return bools
34
55
35
56
def random_string(length=10, upper=False):
36
57
    """
37
58
    Return a string, of specified length, of random lower or uppercase letters.
38
59
    """
39
60
    charset = string.uppercase if upper else string.lowercase
40
61
    return ''.join(random.sample(charset, length))
41
40
62
42
41
def random_postal_address():
63
def random_postal_address():
43
42
    """Return something that looks like a postal address"""
64
    """Return something that looks like a postal address"""
45
43
    return '%s Street' % random.choice(LAST_NAMES)
65
    return '%d %s %s' % (random.randint(1, 100), random.choice(LAST_NAMES),
46
66
        random.choice(STREET_TYPES))
47
44
67
51
45
def random_email_address(first_name, last_name):
68
def random_email_address(
52
46
    """Return something that looks like an email address"""
69
        first_name='', last_name='', company='', address_type='other'):
53
47
    return '%s.%s@%s.com' % (first_name, last_name, random_string(3))
70
    """
54
71
    Return something that looks like an email address.
55
72
    address_type values: 'personal', 'work', 'other'.
56
73
    """
57
74
    # avoid including the dot if one or both names are missing
58
75
    pers_name = '.'.join([name for name in (first_name, last_name) if name])
59
76
    return {'personal': pers_name, 'work': company,
60
77
        }.get(address_type, random_string(5)) + '@example.com'
61
48
78
62
49
def random_postal_code():
79
def random_postal_code():
63
50
    """Return something that looks like a postal code"""
80
    """Return something that looks like a postal code"""
65
51
    return '%s12 3%s' % (random_string(2), random_string(2))
81
    return '%s12 3%s' % (random_string(2, True), random_string(2, True))
66
52
82
67
53
def random_phone_number():
83
def random_phone_number():
68
54
    """Return something that looks like a phone number"""
84
    """Return something that looks like a phone number"""
69
@@ -65,11 +95,39 @@
70
65
    # Record schema
95
    # Record schema
71
66
    fielddict = {'record_type': doctype, 'record_type_version': '1.0'}
96
    fielddict = {'record_type': doctype, 'record_type_version': '1.0'}
72
67
    # Names
97
    # Names
75
68
    first_name = random.choice(FIRST_NAMES) + str(maincount)
98
    # at least one of the three will be present
76
69
    last_name = random.choice(LAST_NAMES)
99
    has_first_name, has_last_name, has_company_name = random_bools(3)
77
100
    first_name = (random.choice(FIRST_NAMES) + str(maincount)
78
101
        ) if has_first_name else ''
79
102
    last_name = random.choice(LAST_NAMES) if has_last_name else ''
80
103
    company = (random.choice(COMPANIES) + str(maincount)
81
104
        ) if has_company_name else ''
82
105
    # Address places and types
83
106
    address_places, email_types, phone_places = [], [], []
84
107
    if has_first_name or has_last_name:
85
108
        for (has_it, seq, val) in zip(
86
109
                # at least one of the three will be present
87
110
                random_bools(3),
88
111
                (address_places, email_types, phone_places),
89
112
                ('home', 'personal', 'home')):
90
113
            if has_it:
91
114
                seq.append(val)
92
115
    if has_company_name:
93
116
        for (has_it, seq) in zip(
94
117
                # at least one of the three will be present
95
118
                random_bools(3),
96
119
                (address_places, email_types, phone_places)):
97
120
            if has_it:
98
121
                seq.append('work')
99
122
    for (has_it, seq) in zip(
100
123
            # none of the three may be present
101
124
            random_bools(3, at_least_one_true=False),
102
125
            (address_places, email_types, phone_places)):
103
126
        if has_it:
104
127
            seq.append('other')
105
70
    # Addresses
128
    # Addresses
106
71
    addresses = {}
129
    addresses = {}
108
72
    for address_type in ('home', 'work', 'other'):
130
    for address_type in address_places:
109
73
        addresses[str(uuid.uuid4())] = {
131
        addresses[str(uuid.uuid4())] = {
110
74
            'address1': random_postal_address(), 'address2': '',
132
            'address1': random_postal_address(), 'address2': '',
111
75
            'pobox': '', 'city': random.choice(CITIES),
133
            'pobox': '', 'city': random.choice(CITIES),
112
@@ -77,20 +135,23 @@
113
77
            'country': random.choice(COUNTRIES), 'description': address_type}
135
            'country': random.choice(COUNTRIES), 'description': address_type}
114
78
    # Email addresses
136
    # Email addresses
115
79
    email_addresses = {}
137
    email_addresses = {}
117
80
    for email_address_type in ('home', 'work', 'other'):
138
    for email_address_type in email_types:
118
81
        email_addresses[str(uuid.uuid4())] = {
139
        email_addresses[str(uuid.uuid4())] = {
120
82
            'address': random_email_address(first_name, last_name),
140
            'address': random_email_address(
121
141
                first_name, last_name, company, email_address_type),
122
83
            'description': email_address_type}
142
            'description': email_address_type}
123
84
    # Phone numbers
143
    # Phone numbers
124
85
    phone_numbers = {}
144
    phone_numbers = {}
126
86
    for phone_number_type in ('home', 'work', 'other'):
145
    for phone_number_type in phone_places:
127
87
        phone_numbers[str(uuid.uuid4())] = {
146
        phone_numbers[str(uuid.uuid4())] = {
128
88
            'priority': 0, 'number': random_phone_number(),
147
            'priority': 0, 'number': random_phone_number(),
129
89
            'description': phone_number_type}
148
            'description': phone_number_type}
130
90
    # Store data in fielddict
149
    # Store data in fielddict
131
150
    if (has_first_name or has_last_name) and head_or_tails():
132
151
        fielddict['birth_date'] = random_birth_date()
133
91
    fielddict.update({'first_name': first_name, 'last_name': last_name,
152
    fielddict.update({'first_name': first_name, 'last_name': last_name,
134
92
        'addresses': addresses, 'email_addresses': email_addresses,
153
        'addresses': addresses, 'email_addresses': email_addresses,
136
93
        'phone_numbers': phone_numbers, 'birth_date': random_birth_date()})
154
        'phone_numbers': phone_numbers, 'company': company})
137
94
    # Possibly add example application annotations
155
    # Possibly add example application annotations
138
95
    if app_annots:
156
    if app_annots:
139
96
        fielddict['application_annotations'] = app_annots
157
        fielddict['application_annotations'] = app_annots
Status:	Merged
Approved by:	dobey on 2009-09-04
Approved revision:	46
Merged at revision:	not available
Proposed branch:	lp://qastaging/~teknico/desktopcouch/more-varied-random-contacts
Merge into:	lp://qastaging/desktopcouch
Diff against target:	None lines
To merge this branch:	bzr merge lp://qastaging/~teknico/desktopcouch/more-varied-random-contacts
Related bugs:	Link a bug report
Reviewer	Date Requested	Status
dobey (community)		Approve on 2009-09-04
Eric Casteleijn (community)	2009-09-04	Approve on 2009-09-04
Review via email: mp+11192@code.qastaging.launchpad.net