mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-29 10:13:57 +01:00
22 lines
607 B
Python
22 lines
607 B
Python
|
import asyncio
|
||
|
|
|
||
|
|
async def invalidate_key(provider_and_key):
|
||
|
|
if not provider_and_key:
|
||
|
|
return
|
||
|
|
|
||
|
|
provider = provider_and_key.split('>')[0]
|
||
|
|
provider_file = f'secret/{provider}.txt'
|
||
|
|
key = provider_and_key.split('>')[1]
|
||
|
|
|
||
|
|
with open(provider_file, encoding='utf8') as f_in:
|
||
|
|
text = f_in.read()
|
||
|
|
|
||
|
|
with open(provider_file, 'w', encoding='utf8') as f_out:
|
||
|
|
f_out.write(text.replace(key, ''))
|
||
|
|
|
||
|
|
with open(f'secret/{provider}.invalid.txt', 'a', encoding='utf8') as f:
|
||
|
|
f.write(key + '\n')
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
asyncio.run(invalidate_key('closed>sk-...'))
|