Jelajahi Sumber

Added 'spaces' option.

Taddeus Kroes 13 tahun lalu
induk
melakukan
9e8cbbdc3e
1 mengubah file dengan 5 tambahan dan 0 penghapusan
  1. 5 0
      generate.py

+ 5 - 0
generate.py

@@ -296,6 +296,8 @@ if __name__ == '__main__':  # pragma: nocover
     parser.add_argument('-a', '--create-all', action='store_true',
                         help='create all available options and variables')
     parser.add_argument('-d', '--dir', help='directory to save model files in')
+    parser.add_argument('-s', '--spaces', nargs='?', const=4, type=int,
+                        help='use spaces instead of tabs (4 by default)')
     args = parser.parse_args()
     options = {}
 
@@ -306,6 +308,9 @@ if __name__ == '__main__':  # pragma: nocover
     if args.create_all:
         options['create_select'] = options['create_accessible'] = True
 
+    if args.spaces:
+        TAB = args.spaces * ' '
+
     conn = mysql.connect(args.host, args.user, args.password, args.dbname)
     models = create_models(conn, options)
     conn.close()