Merge pull request #14 from yangmingming/main

Hide password entry in reset_init script

README.md

@@ -122,7 +122,7 @@ Otherwise, it cannot decode some addresses such as `512057cd4cfa03f27f7b18c2fe45
 ```bash
 pip3 install python-dotenv;
 pip3 install psycopg2-binary;
-python3 -m pip install json5;
+python3 -m pip install json5 stdiomask;
 ```
 
 **Setup .env files and DBs**

modules/bitmap_api/reset_init.py

@@ -1,7 +1,9 @@
 # pip install python-dotenv
+# pip install stdiomask
 
 import os
 from dotenv import dotenv_values
+import stdiomask
 
 init_env = True
 
@@ -50,7 +52,7 @@ if init_env:
     res = input("Bitmap Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("Bitmap Postgres DB password: ")
+    res = stdiomask.getpass("Bitmap Postgres DB password: ")
     DB_PASSWD = res
   res = input("Bitmap Postgres DB use SSL (Default: true) may need to be set to false on Windows machines: ")
   if res != '':

modules/bitmap_index/reset_init.py

@@ -1,7 +1,8 @@
 # pip install python-dotenv
 # pip install psycopg2-binary
+# pip install stdiomask
 
-import os, psycopg2
+import os, psycopg2, stdiomask
 from dotenv import load_dotenv, dotenv_values
 
 init_env = True
@@ -56,7 +57,7 @@ if init_env:
     res = input("Bitmap Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("Bitmap Postgres DB password: ")
+    res = stdiomask.getpass("Bitmap Postgres DB password: ")
     DB_PASSWD = res
   use_main_env = False
   main_env_exists = os.path.isfile('../main_index/.env')
@@ -85,7 +86,7 @@ if init_env:
     res = input("Main Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_METAPROTOCOL_DATABASE = res
-    res = input("Main Postgres DB password: ")
+    res = stdiomask.getpass("Main Postgres DB password: ")
     DB_METAPROTOCOL_PASSWD = res
     res = input("Network type (Default: mainnet) options: mainnet, testnet, signet, regtest: ")
     if res != '':
@@ -177,4 +178,4 @@ for sql in sqls:
 cur.close()
 conn.close()
 
-print('done')
+print('done')

modules/brc20_api/reset_init.py

@@ -1,7 +1,9 @@
 # pip install python-dotenv
+# pip install stdiomask
 
 import os
 from dotenv import load_dotenv, dotenv_values
+import stdiomask
 
 init_env = True
 
@@ -52,7 +54,7 @@ if init_env:
     res = input("BRC20 Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("BRC20 Postgres DB password: ")
+    res = stdiomask.getpass("BRC20 Postgres DB password: ")
     DB_PASSWD = res
     res = input("Use extra tables (Default: true): ")
     if res != '':

modules/brc20_index/reset_init.py

@@ -1,7 +1,8 @@
 # pip install python-dotenv
 # pip install psycopg2-binary
+# pip install stdiomask
 
-import os, psycopg2
+import os, psycopg2, stdiomask
 from dotenv import load_dotenv, dotenv_values
 
 init_env = True
@@ -57,7 +58,7 @@ if init_env:
     res = input("BRC20 Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("BRC20 Postgres DB password: ")
+    res = stdiomask.getpass("BRC20 Postgres DB password: ")
     DB_PASSWD = res
   use_main_env = False
   main_env_exists = os.path.isfile('../main_index/.env')
@@ -86,7 +87,7 @@ if init_env:
     res = input("Main Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_METAPROTOCOL_DATABASE = res
-    res = input("Main Postgres DB password: ")
+    res = stdiomask.getpass("Main Postgres DB password: ")
     DB_METAPROTOCOL_PASSWD = res
     res = input("Network type (Default: mainnet) options: mainnet, testnet, signet, regtest: ")
     if res != '':
@@ -194,4 +195,4 @@ if create_extra_tables:
 cur.close()
 conn.close()
 
-print('done')
+print('done')

modules/main_index/reset_init.py

@@ -1,7 +1,8 @@
 # pip install python-dotenv
 # pip install psycopg2-binary
+# pip install stdiomask
 
-import os, psycopg2, pathlib
+import os, psycopg2, pathlib, stdiomask
 from dotenv import load_dotenv
 
 init_env = True
@@ -41,7 +42,7 @@ if init_env:
   res = input("Main Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
   if res != '':
     DB_DATABASE = res
-  res = input("Main Postgres DB password: ")
+  res = stdiomask.getpass("Main Postgres DB password: ")
   DB_PASSWD = res
   res = input("Main Postgres DB use SSL (Default: true) may need to be set to false on Windows machines: ")
   if res != '':
@@ -55,7 +56,7 @@ if init_env:
   res = input("Bitcoin RPC username (Default: (empty)) leave default to use .cookie file: ")
   if res != '':
     BITCOIN_RPC_USER = res
-  res = input("Bitcoin RPC password (Default: (empty)) leave default to use .cookie file: ")
+  res = stdiomask.getpass("Bitcoin RPC password (Default: (empty)) leave default to use .cookie file: ")
   if res != '':
     BITCOIN_RPC_PASSWD = res
   res = input("Ord binary command (Default: ./ord) change to ord.exe on Windows (without ./): ")
@@ -174,4 +175,4 @@ ord_log_file_index_path.write_text("")
 log_file_error_path = pathlib.Path("log_file_error.txt").absolute()
 log_file_error_path.write_text("")
 
-print('done')
+print('done')

modules/sns_api/reset_init.py

@@ -1,6 +1,7 @@
 # pip install python-dotenv
+# pip install stdiomask
 
-import os
+import os, stdiomask
 from dotenv import dotenv_values
 
 init_env = True
@@ -50,7 +51,7 @@ if init_env:
     res = input("SNS Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("SNS Postgres DB password: ")
+    res = stdiomask.getpass("SNS Postgres DB password: ")
     DB_PASSWD = res
   res = input("SNS Postgres DB use SSL (Default: true) may need to be set to false on Windows machines: ")
   if res != '':

modules/sns_index/reset_init.py

@@ -1,7 +1,8 @@
 # pip install python-dotenv
 # pip install psycopg2-binary
+# pip install stdiomask
 
-import os, psycopg2
+import os, psycopg2, stdiomask
 from dotenv import load_dotenv, dotenv_values
 
 init_env = True
@@ -56,7 +57,7 @@ if init_env:
     res = input("SNS Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_DATABASE = res
-    res = input("SNS Postgres DB password: ")
+    res = stdiomask.getpass("SNS Postgres DB password: ")
     DB_PASSWD = res
   use_main_env = False
   main_env_exists = os.path.isfile('../main_index/.env')
@@ -85,7 +86,7 @@ if init_env:
     res = input("Main Postgres DB name (Default: postgres) leave default if no new dbs are created: ")
     if res != '':
       DB_METAPROTOCOL_DATABASE = res
-    res = input("Main Postgres DB password: ")
+    res = stdiomask.getpass("Main Postgres DB password: ")
     DB_METAPROTOCOL_PASSWD = res
     res = input("Network type (Default: mainnet) options: mainnet, testnet, signet, regtest: ")
     if res != '':
@@ -177,4 +178,4 @@ for sql in sqls:
 cur.close()
 conn.close()
 
-print('done')
+print('done')