예제 #1
0
파일: cli.py 프로젝트: amarazov/gamepool
 def games(filename):
     """Initialize the game settings with csv file"""
     with open(filename) as csvfile:
         reader = csv.DictReader(csvfile, dialect='singlequote')
         for row in reader:
             print(row['name'], row['typ'], row['args'])
             game = Game.query.filter_by(name=row['name'],
                                         typ=row['typ']).first()
             if game is None:
                 if row['args'] != 'DELETE':
                     game = Game(name=row['name'],
                                 typ=row['typ'],
                                 args=row['args'])
                     db.session.add(game)
                     star = Game(name=row['name'], typ='*', args='{}')
                     db.session.add(star)
                     db.session.commit()
             else:
                 if row['args'] == 'DELETE':
                     db.session.delete(game)
                 else:
                     game.args = row['args']
                     db.session.add(game)
         db.session.commit()