def subtonews(first_name, last_name, email, chapter_id):
cursor = connection.cursor()
cursor.execute(
'SELECT u.id, u.email_chapter_optin FROM auth_user as u, auth_memberstatus as ms WHERE u.email = "'
+ email +
'" AND u.id = ms.user_id AND ms.status_date_end IS NULL AND ms.statusType_id = 8 AND u.chapter_id = '
+ str(chapter_id))
user = cursor.fetchone()
if user:
if int(user[1]) == 1:
raise SubToNewsException(
_('That email address is already subscribed'))
else:
# reinstate a previous subscriber's subscription
user = User.objects.get(pk=user[0])
user.email_chapter_optin = True
user.first_name = first_name
user.last_name = last_name
user.save()
else:
user = User()
columns = ['first_name', 'last_name', 'email']
row = [first_name, last_name, email]
user.username = generate_unique_username(row, columns)
user.first_name = first_name
user.last_name = last_name
user.email = email
user.chapter_id = chapter_id
user.email_chapter_optin = True
user.date_joined = datetime.now()
user.save()
# Must be called after save() because the primary key
# is required for these
mt = MemberStatus(user_id=user.pk, statusType_id=8)
mt.save()
def importcsv(filerows, welcomeemail, defaults, chapter, updateuser,
ignore_email):
columns = None
users_imported = 0
username_pos = 0
users_updated = 0
existing_users = 0
existing_emails = 0
count = -1
username_field_exists_flag = False
user_already_exists = False
msg = ""
if 'date_joined' not in defaults:
defaults['date_joined'] = datetime.now()
elif defaults['date_joined'] == None:
defaults['date_joined'] = datetime.now()
for row in filerows:
if any(row):
# Create new user
newuser = User()
count += 1
user_already_exists_flag = False
# Get column names from first row, also get the positions of the fields so that we can extract their values
# using their positions later.
if (columns == None):
columns = row
if 'first_name' not in columns:
raise RgImportCsvException(
_('You must specify a first_name field'))
else:
first_name_pos = columns.index('first_name')
if 'last_name' not in columns:
raise RgImportCsvException(
_('You must specify a last_name field'))
else:
last_name_pos = columns.index('last_name')
if 'email' not in columns:
raise RgImportCsvException(
_('You must specify an email field'))
else:
email_pos = columns.index('email')
if 'username' in columns:
username_pos = columns.index('username')
username_field_exists_flag = True
if 'mobile' in columns:
mobile_pos = columns.index('mobile')
continue
# Process row
i = 0
# extracting the values of the username, email, first_name and last_name fields for each row.
if username_field_exists_flag:
uname = row[username_pos]
else:
uname = ''
email = row[email_pos]
first_name = row[first_name_pos]
last_name = row[last_name_pos]
# now remove all the whitespaces from the extracted values.
uname_data = uname.strip()
email_data = email.strip()
first_name_data = first_name.strip()
last_name_data = last_name.strip()
# check if any of the values is None or empty for a row. If yes, form an error message and ignore that row.
if first_name_data == None or first_name_data == '':
msg += ("<br>First name not provided for row %d - row ignored."
) % count
continue
if last_name_data == None or last_name_data == '':
msg += ("<br>Last name not provided for row %d - row ignored."
) % count
continue
if email_data == None or email_data == '':
msg += (
"<br>Email not provided for row %d - row ignored.") % count
continue
# check if the username exists, if yes, check if the 'updateuser' checkbox is ticked. If it is ticked,
# then get the row with the matching username (and, as we will see, replace its contents). Otherwise, ignore.
# Also, they must be from the same chapter
if not check_username(uname_data):
user_already_exists_flag = True
if updateuser:
newuser = User.objects.get(username=uname_data)
if newuser.chapter == chapter:
existing_users += 1
else:
msg += (
"<br>Row %d has a username clash (%s) with another chapter - row ignored"
) % (count, uname_data)
continue
else:
msg += (
"<br>Row %d has a username clash (%s) - row ignored"
) % (count, uname_data)
continue
# check if the email exists for any user, if yes, check if the 'ignore_email' checkbox is ticked. If it is not ticked,
# then get the row with the matching username (and, as we will see, replace its contents). Otherwise, ignore.
# Also, they must be from the same chapter
elif not check_email_and_chapter(email_data, chapter):
existing_emails += 1
if ignore_email:
msg += (
"<br>Row %d's email address (%s) matches an existing user - row ignored"
) % (count, email_data)
continue
for cell in row:
colname = columns[i]
if colname == 'first_name':
stringval(colname, cell, newuser, defaults)
elif colname == 'last_name':
stringval(colname, cell, newuser, defaults)
elif colname == 'email':
stringval(colname, cell, newuser, defaults)
elif colname == 'username':
data = cell.strip()
if data != "":
new_username = data
else:
new_username = generate_unique_username(row, columns)
newuser.username = new_username
elif colname == 'password':
data = cell.strip()
if data != "":
plaintext_password = data
else:
plaintext_password = User.objects.make_random_password(
6)
newuser.set_password(plaintext_password)
elif colname == 'alt_email':
stringval(colname, cell, newuser, defaults)
elif colname == 'mobile':
num = cell.strip().replace(' ', '').replace('+', '')
if num != '':
regexes = MobileRegex.objects.filter(
collection=chapter.mobile_regexes)
try:
number_valid = False
for regex in regexes:
matches = re.compile(regex.regex).findall(num)
if matches == []:
matches = re.compile(
regex.regex).findall("0" + num)
if matches == []:
continue
else:
num = "0" + num
num = regex.prepend_digits + num[regex.
strip_digits:]
number_valid = True
except ValueError:
number_valid = False
if number_valid:
newuser.mobile = num
elif colname == 'date_joined':
dateval(colname, cell, newuser, defaults)
elif colname == 'dob':
dateval(colname, cell, newuser, defaults)
elif colname == 'gender':
numval(colname, cell, newuser, defaults, [0, 1, 2])
elif colname == 'course':
stringval(colname, cell, newuser, defaults)
elif colname == 'uni_start':
dateval(colname, cell, newuser, defaults)
elif colname == 'uni_end':
dateval(colname, cell, newuser, defaults)
elif colname == 'university_id':
unis = University.objects.all()
uni_ids = [-1]
for uni in unis:
uni_ids.append(uni.pk)
numval(colname, cell, newuser, defaults, uni_ids)
if getattr(newuser, 'university_id', 0) == -1:
newuser.university_id = chapter.university_id
elif colname == 'course_type':
numval(colname, cell, newuser, defaults, [1, 2])
elif colname == 'student_type':
numval(colname, cell, newuser, defaults, [1, 2])
elif colname == 'student_number':
stringval(colname, cell, newuser, defaults)
elif colname == 'privacy':
numval(colname, cell, newuser, defaults, [0, 5, 10, 20])
elif colname == 'dob_public':
boolval(colname, cell, newuser, defaults)
elif colname == 'email_public':
boolval(colname, cell, newuser, defaults)
elif colname == 'email_chapter_optin':
boolval(colname, cell, newuser, defaults)
elif colname == 'mobile_marketing_optin':
boolval(colname, cell, newuser, defaults)
elif colname == 'email_reminder_optin':
boolval(colname, cell, newuser, defaults)
elif colname == 'mobile_reminder_optin':
boolval(colname, cell, newuser, defaults)
elif colname == 'email_newsletter_optin':
boolval(colname, cell, newuser, defaults)
elif colname == 'email_careers_newsletter_AU_optin':
boolval(colname, cell, newuser, defaults)
else:
pass # Unknown column, ignore
# Increment column and do the loop again
i += 1
# If we still don't have a username and/or password
# by this stage, let's generate one
if getattr(newuser, 'username', '') == '':
new_username = generate_unique_username(row, columns)
newuser.username = new_username
if getattr(newuser, 'password', '') == '':
plaintext_password = User.objects.make_random_password(6)
newuser.set_password(plaintext_password)
# And finally...
newuser.chapter = chapter
newuser.save()
# If updating an existing user, we don't need to do the rest
if user_already_exists_flag:
continue
# Should be the default at the model-level,
# but just to be sure...
newuser.is_active = True
newuser.is_staff = False
newuser.is_superuser = False
# Apply any unapplied defaults
for key, value in defaults.iteritems():
if key not in columns:
setattr(newuser, key, value)
newuser.save()
# Must be called after newuser.save() because the primary key
# is required for these
mt = MemberStatus(user_id=newuser.pk,
statusType_id=1,
status_date_start=newuser.date_joined)
mt.save()
# Send welcome email
if welcomeemail:
message = EmailMessage()
try:
message.subject = welcomeemail['subject'].format(
chapter=chapter,
user=newuser,
plaintext_password=plaintext_password)
except Exception:
newuser.delete()
raise RgImportCsvException(
_('Welcome email subject format is invalid'))
try:
message.body = welcomeemail['body'].format(
chapter=chapter,
user=newuser,
plaintext_password=plaintext_password)
except Exception:
newuser.delete()
raise RgImportCsvException(
_('Welcome email format is invalid'))
message.from_address = '*****@*****.**'
message.reply_address = '*****@*****.**'
message.from_name = chapter.name
message.sender = User.objects.get(username='******')
message.html = welcomeemail['html']
message.status = -1
message.save()
recipient = EmailRecipient()
recipient.message = message
recipient.user = newuser
recipient.to_name = newuser.get_full_name()
recipient.to_address = newuser.email
recipient.save()
message.status = 0
message.save()
users_imported += 1
return (users_imported, existing_users, existing_emails, msg)
