Added user_regdate to forummanager so everyone isn't from 1970

This commit is contained in:
Raynaldo Rivera 2014-10-13 11:04:40 -07:00
parent 20b9152a1c
commit 9f0047b3a7
2 changed files with 16 additions and 6 deletions

View File

@ -1,4 +1,6 @@
import os import os
import calendar
from datetime import datetime
from passlib.apps import phpbb3_context from passlib.apps import phpbb3_context
from django.db import connections from django.db import connections
@ -6,8 +8,8 @@ from django.db import connections
class ForumManager: class ForumManager:
SQL_ADD_USER = r"INSERT INTO phpbb_users (username, username_clean, " \ SQL_ADD_USER = r"INSERT INTO phpbb_users (username, username_clean, " \
r"user_password, user_email, group_id , user_permissions, " \ r"user_password, user_email, group_id, user_regdate, user_permissions, " \
r"user_sig, user_occ, user_interests) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)" r"user_sig, user_occ, user_interests) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"
SQL_DEL_USER = r"DELETE FROM phpbb_users where username = %s" SQL_DEL_USER = r"DELETE FROM phpbb_users where username = %s"
@ -78,6 +80,12 @@ class ForumManager:
cursor.execute(ForumManager.SQL_GET_USER_GROUPS, [userid]) cursor.execute(ForumManager.SQL_GET_USER_GROUPS, [userid])
return [row[0] for row in cursor.fetchall()] return [row[0] for row in cursor.fetchall()]
@staticmethod
def __get_current_utc_date():
d = datetime.utcnow()
unixtime = calendar.timegm(d.utctimetuple())
return unixtime
@staticmethod @staticmethod
def __create_group(groupname): def __create_group(groupname):
cursor = connections['phpbb3'].cursor() cursor = connections['phpbb3'].cursor()
@ -94,7 +102,6 @@ class ForumManager:
cursor = connections['phpbb3'].cursor() cursor = connections['phpbb3'].cursor()
cursor.execute(ForumManager.SQL_REMOVE_USER_GROUP, [userid, groupid]) cursor.execute(ForumManager.SQL_REMOVE_USER_GROUP, [userid, groupid])
@staticmethod @staticmethod
def add_user(username, email, groups): def add_user(username, email, groups):
cursor = connections['phpbb3'].cursor() cursor = connections['phpbb3'].cursor()
@ -108,8 +115,10 @@ class ForumManager:
ForumManager.__update_user_info(username_clean, email, pwhash) ForumManager.__update_user_info(username_clean, email, pwhash)
else: else:
try: try:
cursor.execute(ForumManager.SQL_ADD_USER, [username_clean, username_clean, pwhash, cursor.execute(ForumManager.SQL_ADD_USER, [username_clean, username_clean, pwhash,
email, 2, "", "", "", ""]) email, 2, ForumManager.__get_current_utc_date(),
"", "", "", ""])
ForumManager.update_groups(username_clean, groups) ForumManager.update_groups(username_clean, groups)
except: except:
pass pass
@ -147,9 +156,9 @@ class ForumManager:
act_groups = set([g.replace(' ', '-') for g in groups]) act_groups = set([g.replace(' ', '-') for g in groups])
addgroups = act_groups - user_groups addgroups = act_groups - user_groups
remgroups = user_groups - act_groups remgroups = user_groups - act_groups
print username
print addgroups print addgroups
print remgroups print remgroups
print userid
for g in addgroups: for g in addgroups:
if not g in forum_groups: if not g in forum_groups:
forum_groups[g] = ForumManager.__create_group(g) forum_groups[g] = ForumManager.__create_group(g)

View File

@ -36,4 +36,5 @@ def deactivate_services(user):
def generate_corp_group_name(corpname): def generate_corp_group_name(corpname):
return 'Corp_'+corpname.replace(' ','_') return 'Corp_'+corpname.replace(' ', '_')