1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22

# TODO: return the rank of user_id
user_id = int(user_id)    
results = contests.aggregate([
  {
    '$group': {
      '_id': '$user_id', 
      'sum_score': {'$sum': '$score'}, 
      'sum_time': {'$sum': '$submit_time'}
    }
  },
  {
    '$sort': {
      'sum_score': -1,
      'sum_time': -1
    }
  }
])
# CRISES: id_list may cause large memory consumption
id_list = [int(value['_id']) for value in results]
# INPROVEMENTS: could use binary search
try: return id_list.index(user_id)+1
except: return 0

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

import sys
import csv

class ScriptError(Exception):
  def __init__(self, value):
    self.value = value
  def __str__(self):
    return repr('ERROR:'+self.value)

def listinfo():
  with open('serverinfo', 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=' ')
    for row in reader:
      print ' '.join(row)

def addinfo(hostip, user, password):
  with open('serverinfo', 'r+') as csvfile:
    reader = csv.reader(csvfile, delimiter=' ')
    contents = []
    new = [hostip, user]
    for row in reader:
      contents.append([row[0], row[1]])
    if new not in contents:
      writer = csv.writer(csvfile, delimiter=' ')
      writer.writerow([hostip, user, password])
    else:
      raise ScriptError('user exists')

def deleteinfo(hostip):
  f = open('serverinfo', 'r')
  lines = f.readlines()
  f.close()
  f = open('serverinfo', 'w')
  for line in lines:
    if hostip not in line:
      f.write(line)
  f.close()

def main():
  if len(sys.argv) < 2:
    print """
    Usage:
    ======
    python server.py [add|delete|list]
    
    add: add info, required parameter below
         -h hostip
         -u user
         -p password
    delete: delete info, required parameter below
         -h hostip
    list:list all info
    """
  elif sys.argv[1] == 'add':
    if (sys.argv[2] == '-h' or sys.argv[2] == 'hostip')   and \
       (sys.argv[4] == '-u' or sys.argv[4] == 'user')     and \
       (sys.argv[6] == '-p' or sys.argv[6] == 'password') and \
       len(sys.argv) == 8:
      addinfo(sys.argv[3], sys.argv[5], sys.argv[7])
    else:
      raise ScriptError('argument error')
  elif sys.argv[1] == 'delete':
    if (sys.argv[2] == '-h' or sys.argv[2] == 'hostip') and len(sys.argv) == 4:
      deleteinfo(sys.argv[3])
    else:
      raise ScriptError('argument error')
  elif sys.argv[1] == 'list':
    listinfo()
  else: raise ScriptError('no such option')

if __name__ == '__main__':
  main()