Skip to content

Commit 9a0596e

Browse files
committed
refactor(database): use modelOptions helper to streamline Sequelize model initialization
1 parent ac704c6 commit 9a0596e

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

src/submodules/database.js

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -24,23 +24,25 @@ let sequelizeInstance = null;
2424
* @param {Sequelize} sequelize
2525
*/
2626
function initModels(sequelize) {
27-
Episode.init(episodeColumns, { sequelize });
28-
Movie.init(movieColumns, { sequelize });
29-
Series.init(seriesColumns, { sequelize });
27+
const modelOptions = (modelName) => ({ sequelize, modelName });
3028

31-
File.init(fileColumns, { sequelize });
32-
Stream.init(streamColumns, { sequelize });
29+
Episode.init(episodeColumns, modelOptions('Episode'));
30+
Movie.init(movieColumns, modelOptions('Movie'));
31+
Series.init(seriesColumns, modelOptions('Series'));
3332

34-
EpisodeFiles.init(episodeFilesColumns, { sequelize });
35-
MovieFiles.init(movieFileColumns, { sequelize });
33+
File.init(fileColumns, modelOptions('File'));
34+
Stream.init(streamColumns, modelOptions('Stream'));
3635

37-
MovieSet.init(movieSetColumns, { sequelize });
38-
SeriesSet.init(seriesSetColumns, { sequelize });
36+
EpisodeFiles.init(episodeFilesColumns, modelOptions('EpisodeFiles'));
37+
MovieFiles.init(movieFileColumns, modelOptions('MovieFiles'));
3938

40-
TrackMovie.init(trackMovieColumns, { sequelize });
41-
TrackEpisode.init(trackEpisodesColumns, { sequelize });
39+
MovieSet.init(movieSetColumns, modelOptions('MovieSet'));
40+
SeriesSet.init(seriesSetColumns, modelOptions('SeriesSet'));
4241

43-
User.init(userColumns, { sequelize });
42+
TrackMovie.init(trackMovieColumns, modelOptions('TrackMovie'));
43+
TrackEpisode.init(trackEpisodesColumns, modelOptions('TrackEpisode'));
44+
45+
User.init(userColumns, modelOptions('User'));
4446
}
4547

4648
/**

0 commit comments

Comments
 (0)