CodingSouls团队项目冲刺(4)-个人概况

团队冲刺第四天:

  1.继续完成登录注册后端代码:

  1 import * as TypeORM from "typeorm";
  2 import Model from "./common";
  3 
  4 declare var syzoj: any;
  5 
  6 import JudgeState from "./judge_state";
  7 import UserPrivilege from "./user_privilege";
  8 import Article from "./article";
  9 
 10 @TypeORM.Entity()
 11 export default class User extends Model {
 12   static cache = true;
 13 
 14   
 15   @TypeORM.Column({ nullable: true, type: "boolean" })@TypeORM.PrimaryGeneratedColumn()
 16   id: number;
 17 
 18   @TypeORM.Index({ unique: true })
 19   @TypeORM.Column({ nullable: true, type: "varchar", length: 80 })
 20   username: string;
 21 
 22   @TypeORM.Column({ nullable: true, type: "varchar", length: 120 })
 23   email: string;
 24 
 25   @TypeORM.Column({ nullable: true, type: "varchar", length: 120 })
 26   password: string;
 27 
 28   @TypeORM.Column({ nullable: true, type: "varchar", length: 80 })
 29   nickname: string;
 30 
 31   @TypeORM.Column({ nullable: true, type: "text" })
 32   nameplate: string;
 33 
 34   @TypeORM.Column({ nullable: true, type: "text" })
 35   information: string;
 36 
 37   @TypeORM.Index()
 38   @TypeORM.Column({ nullable: true, type: "integer" })
 39   ac_num: number;
 40 
 41   @TypeORM.Index()
 42   @TypeORM.Column({ nullable: true, type: "integer" })
 43   submit_num: number;
 44 
 45   @TypeORM.Column({ nullable: true, type: "boolean" })
 46   is_admin: boolean;
 47 
 48   @TypeORM.Index()
 49   is_show: boolean;
 50 
 51   @TypeORM.Column({ nullable: true, type: "boolean", default: true })
 52   public_email: boolean;
 53 
 54   @TypeORM.Column({ nullable: true, type: "boolean", default: true })
 55   prefer_formatted_code: boolean;
 56 
 57   @TypeORM.Column({ nullable: true, type: "integer" })
 58   sex: number;
 59 
 60   @TypeORM.Column({ nullable: true, type: "integer" })
 61   rating: number;
 62 
 63   @TypeORM.Column({ nullable: true, type: "integer" })
 64   register_time: number;
 65 
 66   static async fromEmail(email): Promise<User> {
 67     return User.findOne({
 68       where: {
 69         email: email
 70       }
 71     });
 72   }
 73 
 74   static async fromName(name): Promise<User> {
 75     return User.findOne({
 76       where: {
 77         username: name
 78       }
 79     });
 80   }
 81 
 82   async isAllowedEditBy(user) {
 83     if (!user) return false;
 84     if (await user.hasPrivilege('manage_user')) return true;
 85     return user && (user.is_admin || this.id === user.id);
 86   }
 87 
 88   getQueryBuilderForACProblems() {
 89     return JudgeState.createQueryBuilder()
 90                      .select(`DISTINCT(problem_id)`)
 91                      .where('user_id = :user_id', { user_id: this.id })
 92                      .andWhere('status = :status', { status: 'Accepted' })
 93                      .andWhere('type != 1')
 94                      .orderBy({ problem_id: 'ASC' })
 95   }
 96 
 97   async refreshSubmitInfo() {
 98     await syzoj.utils.lock(['User::refreshSubmitInfo', this.id], async () => {
 99       this.ac_num = await JudgeState.countQuery(this.getQueryBuilderForACProblems());
100       this.submit_num = await JudgeState.count({
101         user_id: this.id,
102         type: TypeORM.Not(1) // Not a contest submission
103       });
104 
105       await this.save();
106     });
107   }
108 
109   async getACProblems() {
110     let queryResult = await this.getQueryBuilderForACProblems().getRawMany();
111 
112     return queryResult.map(record => record['problem_id'])
113   }
114 
115   async getArticles() {
116     return await Article.find({
117       where: {
118         user_id: this.id
119       }
120     });
121   }
122 
123   async getStatistics() {
124     let statuses = {
125       "Accepted": ["Accepted"],
126       "Wrong Answer": ["Wrong Answer", "File Error", "Output Limit Exceeded"],
127       "Runtime Error": ["Runtime Error"],
128       "Time Limit Exceeded": ["Time Limit Exceeded"],
129       "Memory Limit Exceeded": ["Memory Limit Exceeded"],
130       "Compile Error": ["Compile Error"]
131     };
132 
133     let res = {};
134     for (let status in statuses) {
135       res[status] = 0;
136       for (let s of statuses[status]) {
137         res[status] += await JudgeState.count({
138           user_id: this.id,
139           type: 0,
140           status: s
141         });
142       }
143     }
144 
145     return res;
146   }
147 
148   async renderInformation() {
149     this.information = await syzoj.utils.markdown(this.information);
150   }
151 
152   async getPrivileges() {
153     let privileges = await UserPrivilege.find({
154       where: {
155         user_id: this.id
156       }
157     });
158 
159     return privileges.map(x => x.privilege);
160   }
161 
162   async setPrivileges(newPrivileges) {
163     let oldPrivileges = await this.getPrivileges();
164 
165     let delPrivileges = oldPrivileges.filter(x => !newPrivileges.includes(x));
166     let addPrivileges = newPrivileges.filter(x => !oldPrivileges.includes(x));
167 
168     for (let privilege of delPrivileges) {
169       let obj = await UserPrivilege.findOne({ where: {
170         user_id: this.id,
171         privilege: privilege
172       } });
173 
174       await obj.destroy();
175     }
176 
177     for (let privilege of addPrivileges) {
178       let obj = await UserPrivilege.create({
179         user_id: this.id,
180         privilege: privilege
181       });
182 
183       await obj.save();
184     }
185   }
186 
187   async hasPrivilege(privilege) {
188     if (this.is_admin) return true;
189 
190     let x = await UserPrivilege.findOne({ where: { user_id: this.id, privilege: privilege } });
191     return !!x;
192   }
193 
194   async getLastSubmitLanguage() {
195     let a = await JudgeState.findOne({
196       where: {
197         user_id: this.id
198       },
199       order: {
200         submit_time: 'DESC'
201       }
202     });
203     if (a) return a.language;
204 
205     return null;
206   }
207 }
登录注册后端代码

  2.管理员用户

 1 import * as TypeORM from "typeorm";
 2 import Model from "./common";
 3 
 4 @TypeORM.Entity()
 5 export default class UserPrivilege extends Model {
 6   @TypeORM.Index()
 7   @TypeORM.PrimaryColumn({ type: "integer" })
 8   user_id: number;
 9 
10   @TypeORM.Index()
11   @TypeORM.PrimaryColumn({ type: "varchar", length: 80 })
12   privilege: string;
13 }
管理员用户

  3.测试截图

 

posted @ 2020-04-18 18:55  DemonSlayer  阅读(192)  评论(0编辑  收藏  举报