33from fastapi import FastAPI
44from fastapi .middleware .cors import CORSMiddleware
55
6- import platform_scripts .codeforces as codeforces
6+ import platform_scripts .codeforces as cf
7+ import platform_scripts .codechef as chef
8+ import platform_scripts .leetcode as lt
79from leetcode import contest
810
911app = FastAPI ()
1719 allow_headers = ["*" ], # Set this to the HTTP headers you want to allow
1820)
1921
22+
2023@DeprecationWarning
2124@app .get ("/{username}/leetcode/contest" )
2225def get_leetcode_info (username : str ):
2326 if username [0 ] != '@' :
2427 return contest .get_contest_data (username )
2528
29+
2630@DeprecationWarning
2731@app .get ("/{username}/leetcode/contest/basic" )
2832def get_leetcode_info (username : str ):
29- return contest .get_contest_basic (username )
33+ data = contest .get_contest_basic (username )
34+ data ['warning' ] = {"endpoint to be depreciated by end of 2023" }
35+ return data
36+
3037
3138@DeprecationWarning
3239@app .get ("/{username}/leetcode/contest/basic/rank" )
3340def get_leetcode_rank (username : str ):
3441 pass
3542 return contest .get_contest_rank (username )
3643
44+
3745@DeprecationWarning
3846@app .get ("/{username}/leetcode/{query}" )
3947def get_leetcode_info_based_on_query (username : str , query : str ):
4048 query = query .split ("+" )
4149 pass
4250 return contest .get_leetcode_info (username , query )
4351
52+
4453@app .get ("/codeforces/{username}" )
4554def get_codeforces_info (username : str ):
46- return {
47- "rating" : codeforces .get_current_rating (username )
48- }
55+ return cf .get_contest_data (username )
56+
4957
5058@app .get ("/leetcode/{username}" )
5159def get_leetcode_info (username : str ):
52- return {
53- "rating" : contest .get_contest_basic (username )["data" ]["userContestRanking" ]["rating" ]
54- }
60+ return lt .get_contest_data (username )
61+
62+
63+ @app .get ("/codechef/{username}" )
64+ def get_codechef_info (username : str ):
65+ return chef .get_contest_data (username )
66+
5567
5668@app .get ("/status" )
5769def get_status ():
5870 return json .load (open ("status.json" , "r" ))
5971
72+
6073@app .get ("/{username}" )
6174def get_all_ratings_from_username (username : str ):
6275 with open ('users.json' ) as file :
@@ -65,12 +78,29 @@ def get_all_ratings_from_username(username: str):
6578 userdata = users [username ]
6679 else :
6780 return {"error" : "username not found!" ,
68- "code" : 1001 }
81+ "code" : 1100 }
6982
7083 body = {}
7184 if "leetcode" in userdata .keys ():
7285 body ['leetcode' ] = get_leetcode_info (userdata ['leetcode' ]) # TODO: use better functions
7386 if "codeforces" in userdata .keys ():
7487 body ['codeforces' ] = get_codeforces_info (userdata ['codeforces' ]) # TODO: use better functions
88+ if "codechef" in userdata .keys ():
89+ body ['codechef' ] = chef .get_contest_rating (userdata ['codeforces' ]) # TODO: function is ok but not complete
90+
91+ return body
92+
7593
76- return body
94+ @app .get ("/{username}/leetcode" )
95+ def get_leetcode_rating (username : str ):
96+ if username [0 ] == "@" :
97+ with open ('users.json' ) as file :
98+ users : dict = json .load (file )
99+ if username not in users .keys ():
100+ return {"error" : "username not found!" ,
101+ "code" : 1100 }
102+ if "leetcode" not in users [username ].keys ():
103+ return {"error" : f"leetcode id for { username } ,
104+ "code" : 1101 }
105+ username = users [username ]['leetcode' ]
106+ return lt .get_contest_data (username )
0 commit comments