diff --git a/myriad/manage.py b/myriad/manage.py index 576a736..ff5091a 100644 --- a/myriad/manage.py +++ b/myriad/manage.py @@ -255,6 +255,7 @@ def import_groups(groups): def import_member(member): db = get_db() + mid = member["id"] date_raw = member["date-created"].split(",") date = date_raw[0] day,month,year = date.split("/") @@ -273,8 +274,36 @@ def import_member(member): theme = member["theme"] - #db.execute("INSERT INTO member (created,user_id, member_name, bio,public) VALUES (?, ?, ?, ?, ?)",(data[0], data[1], data[2], data[3], data[4])) - #db.commit() + user_id = 0 + + db.execute("INSERT INTO member (id,created,user_id, member_name,subtitle, bio,public,theme) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",(mid, date_created, user_id, name, subtitle, description,privacy, theme)) + db.commit() + + groups = member["groups"] + for group in groups: + db.execute("INSERT INTO group_members (group_id, member_id) VALUES (?, ?)",(group, mid)) + db.commit() + + blog = member["blog"] + for post in blog: + date_raw = post["date-created"].split(",") + date = date_raw[0] + day,month,year = date.split("/") + time = date_raw[1] + hour,minute,second = time.split(":") + date_created = datetime.datetime(int(year), int(month), int(day), int(hour), int(minute), int(second)) + + title = post["title"] + content = post["content"] + + if post["privacy"] == "public": + privacy = 1 + else: + privacy = 0 + + db.execute("INSERT INTO blog (member_id, created, title, content, public) VALUES (?, ?, ?, ?, ?)",(mid, date_created, title, content, privacy)) + db.commit() + @bp.route("/admin", methods=("GET", "POST")) @@ -295,10 +324,13 @@ def admin(): for member in members: import_member(member) - return "go home" elif "zip" in request.files: + file = request.files["zip"] + with zipfile.ZipFile(file, "r") as zipf: + zipf.printdir() + return "upload zip" users = db.execute("SELECT * FROM user").fetchall() @@ -318,6 +350,7 @@ def generate_json(mid): member = db.execute("SELECT * FROM member WHERE id=(?)",(mid,)).fetchone() groups_r = db.execute("SELECT group_id FROM group_members WHERE member_id=(?)",(mid,)).fetchall() blog_r = db.execute("SELECT * FROM blog WHERE member_id=(?)",(mid,)).fetchall() + icons_r = db.execute("SELECT icon_location FROM icons WHERE member_id=(?)",(mid,)).fetchall() if member[9] == 1: privacy = "public" @@ -350,6 +383,10 @@ def generate_json(mid): blog.append(p) + icons = [] + for icon in icons_r: + icons.append(icon[0]) + data = { "id":mid, @@ -361,7 +398,8 @@ def generate_json(mid): "privacy":privacy, "theme":member[10], "groups":groups, - "blog":blog + "blog":blog, + "icons":icons } return data