diff options
author | sjtufs <sjtu_fs@outlook.com> | 2013-08-25 15:45:55 +0800 |
---|---|---|
committer | sjtufs <sjtu_fs@outlook.com> | 2013-08-25 15:45:55 +0800 |
commit | ac7633d8149a28af288ac0b850850cef9b13c151 (patch) | |
tree | 861fe3d57c114b5ec7f3141679428cc6302d21bc /server/piztor/import.py | |
parent | a53f4d21238d2bccd58a22d7485e54f495dbd55d (diff) |
This is alpha
Diffstat (limited to 'server/piztor/import.py')
-rw-r--r-- | server/piztor/import.py | 44 |
1 files changed, 0 insertions, 44 deletions
diff --git a/server/piztor/import.py b/server/piztor/import.py deleted file mode 100644 index 1521849..0000000 --- a/server/piztor/import.py +++ /dev/null @@ -1,44 +0,0 @@ -from sqlalchemy import create_engine -from sqlalchemy.orm import sessionmaker -from model import * - -path = "piztor.sqlite" - -class UserData: - def __init__(self, username, password, sex): - self.username = username - self.password = password - self.sex = sex - -def create_database(): - engine = create_engine('sqlite:///' + path, echo = True) - Base.metadata.drop_all(engine) - Base.metadata.create_all(engine) - -def import_user_data(data): - engine = create_engine('sqlite:///' + path, echo = True) - Session = sessionmaker(bind = engine) - session = Session() - for user in data: - um = UserModel(username = user.username, sex = user.sex) - um.auth = UserAuth(user.password) - um.location = LocationInfo(lat = 0, lng = 0) - session.add(um) - session.commit() - -if __name__ == '__main__': - - from sys import argv, exit - if len(argv) != 2: - print "Usage: " + argv[0] + " FILE" - exit(0) - - data = list() - with open(argv[1], 'r') as f: - while True: - line = f.readline().split() - if len(line) == 0: break - data.append(UserData(line[0], line[1], line[2])) - - create_database() - import_user_data(data) |